Nvidiaドッカーをインストールする方法

次のコマンドを順番に入力します

$ curl -fsSL get.docker.com -o get-docker.sh
$ sudo sh get-docker.sh --mirror Aliyun
$ sudo systemctl enable docker
$ sudo systemctl start docker
$ sudo groupadd docker
$ sudo usermod -aG docker $USER
$ sudo docker run hello-world


$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
$ distribution=$(. /etc/os-release;echo $ID $VERSION_ID)
$ curl -s -L https://nvidia.github.io/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-docker2
$ sudo pkill -SIGHUP dockerd
$ nvidia-docker --version

画像をプルします

$ sudo docker pull pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime        # 这步时间挺长的
$ sudo nvidia-docker run -itd -p 5004:22 pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime

コンテナのIDを問い合わせる

$ sudo docker ps

コンテナにログインする

$ sudo docker exec -it cb52a1489e3a /bin/bash

初期化

$ conda init

インストールされているパッケージを表示する

$ conda list

グラフィックス カード情報の表示

$ nvcc -V

 
 
 

参考リンク：
https://zhuanlan.zhihu.com/p/44962017
https://blog.csdn.net/weixin_43332715/article/details/124414577?spm=1001.2014.3001.5501
https://blog.csdn.net/weixin_43332715 /article/details/124422461?spm=1001.2014.3001.5501
https://blog.csdn.net/weixin_43332715/article/details/124431047
https://hub.docker.com/r/pytorch/pytorch/tags?page=1&ordering =last_updated