Revert "Turn off check for specific CUDA version in conda smokes"

This reverts commit 292217b.
junjihashimoto · Feb 22, 2019 · eb5eea4 · eb5eea4
1 parent bff4947
commit eb5eea4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/smoke_test.sh b/smoke_test.sh
@@ -133,9 +133,8 @@ 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 cudnn)" ]]; then
-    echo "The installed package doesn't seem to be built for CUDA"
-    echo "We expect a cudnn version in the build string"
+  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"
     echo "The full package is $(conda list torch)"
     exit 1
   fi