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 /bin/bash
# 进入容器

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
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值