PyTorch安装

1.查看电脑CUDA版本
CUDA（Compute Unified Device Architecture）是由NVIDIA公司创立的基于他们公司生产的图形处理器GPUs（Graphics Processing Units,可以通俗的理解为显卡）的一个并行计算平台和编程模型。通过CUDA，GPUs可以很方便地被用来进行通用计算（有点像在CPU中进行的数值计算等等）。在没有CUDA之前，GPUs一般只用来进行图形渲染（如通过OpenGL，DirectX）。
（1）打开“控制面板”，找到“硬件和声音”，点击“NVIDIA控制面板”。
（2）点击“NVIDIA控制面板”界面的左下角“系统信息”
（3）在“系统信息”的“组件”中，查看电脑CUDA版本

2.下载CUDA
(1) 搜索CUDA，打开如下网页

（2）按照下图示例，进行下载

（3）默认安装路径

（3）测试安装完成；打开cmd，输入下图中的命令。如果显示是第一个，则需要添加环境变量。

3.下载安装PyTorch
（1）查看python版本，并退出。输入“python”进行版本查看，然后输入“exit()”退出python。
（2）构建虚拟环境

conda create -n pytorch python=3.7.1

（3）激活虚拟环境

activate pytorch

（4）进行安装；当CUDA版本小于9.1时，可以把下列代码中的9.1替换为对应代码。当CUDA版本大于9.1时，可以直接在官网上进行下载。

conda install pytorch torchvision cudatoolkit=9.1 -c pytorch -c defaults -c numba/label/dev

（5）检验是否安装成功；打开Python编辑器，输入下面命令，返回“True”则安装成功

import torch
print(torch.cuda.is_available())