pytorch各种版本最简单安装,不用自己安装cuda cudnn

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的镜像

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
安装PyTorch之前,需要确保你的机器上有NVIDIA GPU并且安装了NVIDIA CUDA工具包和cuDNN库,这是PyTorch使用GPU的前提条件。以下是在Linux系统上安装PyTorch的步骤: 1. 安装CUDA工具包和cuDNN库 首先需要安装NVIDIA CUDA Toolkit和cuDNN库。这里以Ubuntu 16.04为例: ``` # 安装CUDA Toolkit 10.1 $ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin $ sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600 $ wget https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda-repo-ubuntu1604-10-1-local-10.1.105-418.39_1.0-1_amd64.deb $ sudo dpkg -i cuda-repo-ubuntu1604-10-1-local-10.1.105-418.39_1.0-1_amd64.deb $ sudo apt-key add /var/cuda-repo-10-1-local-10.1.105-418.39/7fa2af80.pub $ sudo apt-get update $ sudo apt-get install cuda # 安装cuDNN库 $ wget https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb $ sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb $ sudo apt-get update $ sudo apt-get install libcudnn7 ``` 2. 安装PyTorch 安装PyTorch的最简单方法是使用conda包管理器。如果你还没有安装conda,请先安装conda。 ``` # 创建conda环境 $ conda create --name pytorch python=3.6 # 激活conda环境 $ conda activate pytorch # 安装PyTorch $ conda install pytorch torchvision cudatoolkit=10.1 -c pytorch ``` 安装完成后,可以通过以下代码测试是否安装成功: ```python import torch if torch.cuda.is_available(): device = torch.device("cuda") print("CUDA is available") else: device = torch.device("cpu") print("CUDA is not available") ``` 如果输出结果为“CUDA is available”,则说明PyTorch安装成功并且可以使用GPU加速。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值