diff --git a/.circleci/config.yml b/.circleci/config.yml index a84b1eef2e8..9ab304630d9 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -97,7 +97,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults sudo pip -q install awscli==1.16.35 if [ -n "${CUDA_VERSION}" ]; then - DRIVER_FN="NVIDIA-Linux-x86_64-410.104.run" + DRIVER_FN="NVIDIA-Linux-x86_64-430.40.run" wget "https://s3.amazonaws.com/ossci-linux/nvidia_driver/$DRIVER_FN" sudo /bin/bash "$DRIVER_FN" -s --no-drm || (sudo cat /var/log/nvidia-installer.log && false) nvidia-smi