Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update CUDA driver version in cloud init script #598

Merged
merged 5 commits into from
Dec 8, 2023
Merged

Update CUDA driver version in cloud init script #598

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
fa3e4f3
Merge pull request #585 from mlcommons/dev
priyakasimbeg Nov 18, 2023
af601e9
Merge pull request #587 from mlcommons/dev
priyakasimbeg Nov 22, 2023
7598e02
Merge pull request #591 from mlcommons/dev
fsschneider Nov 28, 2023
ca87833
Merge pull request #593 from mlcommons/dev
priyakasimbeg Nov 28, 2023
db37952
Update cloud-startup.sh
priyakasimbeg Dec 7, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions docker/scripts/cloud-startup.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,10 @@

# Overwrite NVIDIA driver versions and reinstall NVIDIA drivers for CUDA 11.7
sudo chmod 777 /opt/deeplearning/driver-version.sh
sudo echo "export DRIVER_VERSION=515.65.01" > /opt/deeplearning/driver-version.sh
sudo echo "export DRIVER_UBUNTU_DEB="nvidia-driver-local-repo-ubuntu1804-515.65.01_1.0-1_amd64.deb"" >> /opt/deeplearning/driver-version.sh
sudo echo "export DRIVER_VERSION=535.104.05" > /opt/deeplearning/driver-version.sh
sudo echo "export DRIVER_UBUNTU_DEB="nvidia-driver-local-repo-ubuntu1804-535.104.05_1.0-1_amd64.deb"" >> /opt/deeplearning/driver-version.sh
sudo echo "export DRIVER_UBUNTU_CUDA_VERSION="11.7.1"" >> /opt/deeplearning/driver-version.sh
sudo echo "export DRIVER_UBUNTU_PKG=nvidia-driver-515" >> /opt/deeplearning/driver-version.sh
sudo echo "export DRIVER_UBUNTU_PKG=nvidia-driver-535" >> /opt/deeplearning/driver-version.sh
sudo /opt/deeplearning/install-driver.sh

# Install NVIDIA Container Toolkit
Expand Down Expand Up @@ -41,4 +41,4 @@ sleep 30

# Pull latest algorithmic efficiency image
echo "Pulling docker image"
docker pull us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/base_image:latest
docker pull us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo/base_image:latest
Loading