pytorch gpu版本安装最新

pytorch gpu版本安装最新

之前写的一篇时间已久， 正好要安装最新的，今天分享一下

1. 创建虚拟环境

支持下面任一环境：

1. 使用conda
2. 使用python对应的虚拟环境管理工具等 virtualenv
3. 主环境
   这步具体步骤跳过

2. 使用conda来实践

1. 环境

1. 使用conda来创建名叫pytorch190 python版本为3.9的虚拟环境

conda create -n pytorch190 python=3.9

2. 查看本电脑的gpu

1. 设备管理查看显卡版本
   
2. 到英伟达驱动官网查看对应的
   驱动官网
   我自己的驱动版本
   
   查询结果 我自己的是511.79
   
   到英伟达toolkit网站查看自己版本的compoets版本以及自己的CUDA 版本
   

知道自己的cuda版本之后，就可以安装gpu版本的torch了

3. 安装torch

1. 到pytorch官网
   pytorch官网
   选择自己cuda对应最新版本的pytorch版本，我选择了最新的1.9.0版本

2. 进入虚拟环境执行安装命令

安装方式1 conda安装

进入虚拟环境
activate pytorch190
安装方式1
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 -c pytorch

注意

注意 -c pytorch不能少，少了安装的时gpu版本

安装方式2 pip安装

进入虚拟环境
activate pytorch190
安装方式1
pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

注意

这里不能使用清华源,如果时清华源,一定要删除，不然装的时cpu版本的，这里一定要注意。

查看channel
conda config --show
conda config --get channels

# 添加源
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
#pytorch
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/

# 删除源
conda config --remove channels CHANNEL

总结

以上两种方式安装都是国外下载，好几个G很慢
我解决的办法是，自己到 https://download.pytorch.org/whl/torch_stable.html
pytorch官方库下载

下载对应的版本 然后cmd pip install 手动安装即可

测试安装成功

import torch
torch.cuda.is_available()
True