pytorch各种版本 pip 安装命令 查看官网
https://pytorch.org/get-started/previous-versions/
在这里插入图片描述

conda +pytorch 安装
1、安装conda,
2、创建并并激活虚拟环境
- conda create -n pytorch_1.7 python=3.7
- conda activate pytorch_1.7
3、虚拟环境中 pip 安装想要的版本  
  不用先把cuda cudnn 先安装了,如下会自动安装好
  3.1  pip 命令安装(有时候下载超时不行用3.2离线安装方式)
  复制pip  命令
  pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

4、查看pytorch 版本和对应cuda cudnn 版本是不是正确
python
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())

docker + paddle + pytorch

注意 paddle 和 pytorch cuda版本支持一致

1、paddle 官网下载 打包好的镜像
https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/linux-pip.html
在这里插入图片描述

2、安装流程

Paddle docker 2.3.2官网镜像容器安装(最新的按照官方命令替换)
(1)下载镜像:
nvidia-docker pull registry.baidubce.com/paddlepaddle/paddle:2.3.2-gpu-cuda11.2-cudnn8.2-trt8.0
(2)建立容器

  • nvidia-docker run -it --name paddle_2.3.2 --gpus all --shm-size 32g -v /home/syy:/home/syy paddlepaddle/paddle:2.3.2-gpu-cuda11.2-cudnn8 /bin/bash

(3)进入容器
docker exec -it paddle_2.3.2 /bin/bash

(4)安装pytorch (找cuda 11.2的包 pip,)
pytorch 官网没有cuda11.2的包,所以不能安装,官方有cuda11.3 11.6,所以想兼容到一个环境上述paddle 要下载 cuda11.6的镜像

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐