Docker安装pytorch环境(Linux Red Hat系统)
- 下载pytorch镜像 官方镜像地址
sudo docker pull pytorch/pytorch:2.3.1-cuda12.1-cudnn8-devel
#镜像重命名
sudo docker tag name1 name2
- 创建容器
#创建并运行容器
sudo docker run --gpus all -dit -p 容器端口号:宿主机端口号 -v 地址1:地址2 镜像名或ID
sudo docker pull pytorch/pytorch:2.3.1-cuda12.1-cudnn8-devel
#镜像重命名
sudo docker tag name1 name2
#创建并运行容器
sudo docker run --gpus all -dit -p 容器端口号:宿主机端口号 -v 地址1:地址2 镜像名或ID