pass env variable through

junjihashimoto · Feb 3, 2019 · e4650cf · e4650cf
1 parent 91a58bf
commit e4650cf
Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh
@@ -271,6 +271,7 @@ else
 	export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit >=8.0,<8.1"
     else
 	echo "unhandled desired_cuda: $desired_cuda"
+	exit 1
     fi
 
     build_string_suffix="cuda${CUDA_VERSION}_cudnn${CUDNN_VERSION}_${build_string_suffix}"

diff --git a/conda/pytorch-1.0.1/meta.yaml b/conda/pytorch-1.0.1/meta.yaml
@@ -37,6 +37,7 @@ build:
   script_env:
     - CUDA_VERSION
     - CUDNN_VERSION
+    - CONDA_CUDATOOLKIT_CONSTRAINT
     - NO_CUDA
     - CMAKE_ARGS
     - EXTRA_CAFFE2_CMAKE_FLAGS