一、拉取镜像创建pytorch容器

docker pull pytorch/pytorch:1.11.0-cuda11.3-cudnn8-devel
#也可以从下面网址选择更合适版本去下载
https://hub.docker.com/r/pytorch/pytorch/tags?page=1

设置Docker

curl https://get.docker.com | sh \
  && sudo systemctl --now enable docker

设置NVIDIA Container Toolkit

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
      && curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | sudo apt-key add - \
      && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

sudo apt-key adv --keyserver keyserver.ubuntu.com --recv-keys F60F4B3D7FA2AF80

详细解决办法

上面的操作用来解决GPG签名问题，把后面的公钥填成自己报错的公钥

创建pytorch容器

docker run -itd --init --gpus all --ipc=host --name pytorch -p 38:22 --privileged=true pytorch/pytorch:1.11.0-cuda11.3-cudnn8-devel

二、修改容器中的环境

# 进入容器
docker exec [容器名] it bash
apt-get update
apt-get install vim
apt-get install openssh-server
# 修改ssh连接配置，把PermitRootLogin prohibit-passwd改为PermitRootLogin yes
vim /etc/ssh/sshd_config 
passwd root # 修改容器密码
service ssh start # 开启ssh服务

三、pycharm使用ssh连接容器环境出现的问题

如果需要使用控制台运行代码，但是连接远程环境显示python not found有时由于某种原因，Python可执行文件的软链接被中断，或者没有指定到实际的python

ln -s /usr/bin/python [使用 which python 查看python的位置然后填进来，注意没有外面的中括号]