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."