Created
July 20, 2022 19:34
-
-
Save hcho3/c446d97fb385bd76c189641bca20589f to your computer and use it in GitHub Desktop.
Set up NVIDIA docker on Amazon Linux 2
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/bin/bash | |
sudo yum groupinstall -y "Development tools" | |
sudo yum install -y kernel-devel-$(uname -r) | |
aws s3 cp --recursive s3://ec2-linux-nvidia-drivers/latest/ . | |
chmod +x NVIDIA-Linux-x86_64*.run | |
sudo CC=/usr/bin/gcc10-cc ./NVIDIA-Linux-x86_64*.run --silent | |
sudo amazon-linux-extras install docker | |
sudo systemctl --now enable docker | |
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \ | |
&& curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.repo | sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo | |
sudo yum clean expire-cache | |
sudo yum install -y nvidia-docker2 | |
sudo systemctl restart docker |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Path to MSBuild with MSVC 2019:
C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\MSBuild\Current\Bin\MSBuild.exe