diff --git a/ci/test/notebooks.sh b/ci/test/notebooks.sh index 372feea2..de7b5ce9 100644 --- a/ci/test/notebooks.sh +++ b/ci/test/notebooks.sh @@ -10,7 +10,7 @@ source /opt/conda/bin/activate rapids # PyTorch is intentionally excluded from our Docker images due # to its size, but some notebooks still depend on it. case "${CUDA_VER}" in -"10.1" | "10.2") +"10.1" | "10.2" | "11.0") conda install -y -c pytorch "pytorch>=1.4" ;; *)