From ba161d41091a829497f10d4d241fc3525882718c Mon Sep 17 00:00:00 2001 From: dillon-cullinan Date: Fri, 16 Oct 2020 09:14:57 -0700 Subject: [PATCH] FIX Pin pytorch to avoid incorrect solving --- ci/test/notebooks.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ci/test/notebooks.sh b/ci/test/notebooks.sh index da8cada9..372feea2 100644 --- 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") - conda install -y -c pytorch pytorch + conda install -y -c pytorch "pytorch>=1.4" ;; *) echo "Unsupported CUDA version for pytorch."