diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh
index 50f3ca8c4..345b3c3e0 100755
--- 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
index 7dce440bb..986c67598 100644
--- 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