diff --git a/smoke_test.sh b/smoke_test.sh
index 510fcec0b..6bacd16e5 100755
--- 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