pytorch(GPU) 安装简记—Windows版本

1. 安装前准备

①：安装 Anaconda （强烈推荐！！！）

地址：anaconda：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

minicondadi：https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/

②：安装符合你显卡cuda和cudnn

若以前使用配置过GPU，可以忽略此步骤。

2. 开始安装

①：使用conda创建虚拟环境

命令：conda create --name torch python==3.6.X (注意：这里一定要高于3.6.0版本，否则后期安装pytorch报错！！！)

安装成功后激活环境：activate torch （以后都在该环境里操作）

②：查看你的cuda版本（图示如下）

命令：nvcc -V

image-20200622170823193.png

③：查看pytorch的whl文件

pytorch官网：https://pytorch.org/

image-20200622171340125.png

如果你的cuda版本是上图这些，或者是CPU版本，直接参见上图选择下载命令。

否则：

若你的cuda是其他版本，请点击上图绿框，选择满足你cuda的pytorch版本。

④：下载pytorch的whl文件

查看的历史pytorch版本，如下图所示：

image-20200622173337471.png

我的cuda版本是9.0，所以复制上图红色框的连接地址并打开（注意根据自己的cuda版本去选地址），如下图示：（选择点击下载）

image-20200622174008120.png

注意：你需要注意你的python版本，和Windows版本是64位。

⑤：安装pytorch

首先：cd 到下载文件的目录

然后：命令：pip install your_pytorch_file_name

⑥：安装torchvision

下载链接同样是④那个页面中的，图示如下：

image-20200622175232701.png

首先：cd 到下载文件的目录

然后：命令：pip install your_pytorchvision_file_name

3. 安装成功

检查是否安装成功，如果出现下图说明安装成功，一切ok！

image-20200622190347113.png

4. 注意事项

1，python 环境不可 3.6.0,

2，选择下载自己cuda对应的pytorch版本