Merge pull request pytorch#272 from pjh5/smoke_fix

Turn off check for specific CUDA version in conda smokes
junjihashimoto · Feb 22, 2019 · 996c74c · 996c74c
2 parents 6ccafde + 292217b
commit 996c74c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/smoke_test.sh b/smoke_test.sh
@@ -132,8 +132,9 @@ if [[ "$PACKAGE_TYPE" == 'conda' ]]; then
       echo "The installed package is built for CUDA:: $(conda list torch)"
       exit 1
     fi
-  elif [[ -z "$(conda list torch | grep -o cuda$cuda_dot)" ]]; then
-    echo "The installed package doesn't seem to be built for CUDA $cuda_dot"
+  elif [[ -z "$(conda list torch | grep -o cudnn)" ]]; then
+    echo "The installed package doesn't seem to be built for CUDA"
+    echo "We expect a cudnn version in the build string"
     echo "The full package is $(conda list torch)"
     exit 1
   fi