Merge pull request #233 from ajschmidt8/pytorch-version

Update pytorch version
rapidsai · Mar 23, 2021 · 1ea6965 · 1ea6965
2 parents faee43d + de114f6
commit 1ea6965
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ci/test/notebooks.sh b/ci/test/notebooks.sh
@@ -11,7 +11,7 @@ source /opt/conda/bin/activate rapids
 # to its size, but some notebooks still depend on it.
 case "${CUDA_VER}" in
 "10.1" | "10.2" | "11.0")
-    conda install -y -c pytorch "pytorch>=1.4"
+    conda install -y -c pytorch "pytorch=1.7"
     ;;
 *)
     echo "Unsupported CUDA version for pytorch."