coosld/qubic-docker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新nvidia docker container

Browse Source
This commit is contained in:
hyh 2024-06-12 12:52:42 +08:00
parent 0c8a257ef5
commit bc13f60951
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
itgpt-setup.sh
View File

@ -247,8 +247,11 @@ if [[ ! -z "$NVIDIA_PRESENT" ]]; then
else
echo "nvidia-docker 似乎没有启用。正在进行安装..."
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
# curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add
# curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
# nvidia.github.io <- nvidia-docker.geekery.cn
curl -s -L https://nvidia-docker.geekery.cn/nvidia-docker/gpgkey | sudo apt-key add
curl -s -L https://nvidia-docker.geekery.cn/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker
sudo docker run --rm --gpus all $images nvidia-smi