windows安装pytorch1.8-GPU版本

前言

写给刚入坑的小伙伴，跟着来，几分钟就懂

一、安装pytorch1.8 GPU版本

1.安装缺少组件（目前来说少部分电脑需要）

下载360，打开工具箱，打开人工服务，输入“缺少”，然后下方出现的六个都下，一个个下

2.安装VS（只弄python的可以放一边，cocoapi也需要这个）

3.安装cuda

查看目前Version，可以适当去官网更新驱动

一般来说保证cuda=10.2以上都可以安装pytorch1.8

4.安装anaconda

我是安装anaconda2019.10版本

去这里找到对应版本的windows X64版本

然后打开anaconda安装vscode

5.新建虚拟环境

打开anaconda

名字随你取，python选3.7

点击右边▷，用终端打开

永久添加pip的清华源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

优先安装numpy

pip install numpy

6.安装pytorch

有两种方法，一种在线，一种离线安装

1、先介绍在线安装，在线安装，只要进入下面网址，或者直接进入环境输入以下命令行即可

或者直接输入命令行：

conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch

网不好有可能会安装失败，建议还是下方的离线安装，只要找到官方的对应版本即可（对应版本就是在上方的Pytorch官网查看）

2、离线安装

复制下载链接地址到迅雷下

放进一个文件夹

重复第五步骤的用终端打开（也可以直接cmd，然后手动激活环境 ：activate 你的环境命名）

cd 进入上面的文件夹中（如果是D盘，先输入“D:”）

pip install torch-1.8.1+cu111-cp37-cp37m-win_amd64.whl
pip install torchvision-0.9.1+cu111-cp37-cp37m-win_amd64.whl
pip install torchaudio-0.8.1-cp37-none-win_amd64.whl

复制上面直接输入到cmd中

7.验证是否安装成功

上述的终端环境中输入

python

再输入

# TEST
import torch
from torch.backends import cudnn

x = torch.Tensor([1.0])
xx = x.cuda()
print("torch版本:", torch.__version__)
print("torch_cudatoolkit版本:", torch.version.cuda)
print("torch_cuda_可用:", torch.cuda.is_available())
print("torch_cuda_计算:", xx)
print("torch_cudnn_可用:", cudnn.is_acceptable(xx))

显示如下即成功

然后退出python编辑就是“Ctrl + z”

总结：

以上就是今天要讲的内容，本文仅仅简单介绍了windows下安装GPU版本的pytorch环境。