From de114f69defb41dfcbf3257dffeaa803091dc792 Mon Sep 17 00:00:00 2001 From: AJ Schmidt Date: Tue, 23 Mar 2021 12:44:25 -0400 Subject: [PATCH] update pytorch version --- 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 ceb987de..7cd72af0 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" | "11.0") - conda install -y -c pytorch "pytorch>=1.4" + conda install -y -c pytorch "pytorch=1.7" ;; *) echo "Unsupported CUDA version for pytorch."