diff --git a/.circleci/config.yml b/.circleci/config.yml index b98224b82f4..fe1c5aa96df 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -106,7 +106,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-440.59.run" + DRIVER_FN="NVIDIA-Linux-x86_64-460.39.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 @@ -122,7 +122,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults command: | set -e - export pyTorchDockerImageTag=291 + export pyTorchDockerImageTag=9de29bef4a5dc0dd1dd19428d83e5a66a44a1ed2 echo "PyTorchDockerImageTag: "${pyTorchDockerImageTag} cat >/home/circleci/project/ci_build_script.sh \<