Skip to content

Instantly share code, notes, and snippets.

@hcho3
Created July 20, 2022 19:34
Show Gist options
  • Save hcho3/c446d97fb385bd76c189641bca20589f to your computer and use it in GitHub Desktop.
Save hcho3/c446d97fb385bd76c189641bca20589f to your computer and use it in GitHub Desktop.
Set up NVIDIA docker on Amazon Linux 2
#!/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
@hcho3
Copy link
Author

hcho3 commented Jul 22, 2022

Path to MSBuild with MSVC 2019: C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\MSBuild\Current\Bin\MSBuild.exe

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment