pytorch docker部署

windows安装

linux docker下部署

1、创建容器

docker run --name xxrqxx --gpus '"device=2,3"' -m 96g --cpuset-cpus=2-17 -p 50001:22 -p 50002:8888 --restart=always -itd xxjxxx:v1 /bin/bash
# 映射GPU（第2,3张显卡，编号0,1,2,3）、限制CPU16核、映射端口50001、挂掉自动重启
docker exec -it xxrqxx
# 进入容器

2、安装CUDA

3、查看CUDA

nvidia-smi  # nvidia-smi的版本大于nvcc -V的版本
nvcc -V  # 查看是否安装CUDA
或者 /usr/local/cuda/bin/nvcc -V

4、安装anaconda3

sh Anaconda3-2021.11-Linux-x86_64.sh

5、安装pytorch
下载地址：http://download.pytorch.org/whl/torch_stable.html （找到对应版本复制链接使用迅雷下载更快）
下载 torch 1.12.1+cu113 和 torchvision 0.13.1+cu113

pip install torch-1.12.1+cu113-cp39-cp39-linux_x86_64.whl
pip install torchvision-0.13.1+cu113-cp39-cp39-linux_x86_64.whl