Windows下安装Pytorch过程记录

发表于 | 评论数： | 阅读次数：

PyTorch是一个开源的Python机器学习库，基于Torch，底层由C++实现，应用于人工智能领域，如自然语言处理。它最初由Facebook的人工智能研究团队开发，并且被用于Uber的概率编程软件Pyro。

PyTorch主要有两大特征：

  • 类似于NumPy的张量计算，可使用GPU加速;
  • 基于带自动微分系统的深度神经网络;

下面记录在Windows操作系统下安装PyTorch的方法。

安装Anaconda

到Anaconda Distribution下载合适的Anaconda。

查看cuda版本

参考Windows系统查看CUDA版本号。例如本机cuda版本为9.2。

安装pytorch

进入pytorch官网的GET STARTED，根据自身计算机环境，选择PyTorch Build、Your OS、Package、Language和CUDA，即得到安装pytorch的命令：

1
2
3
4
// 基于Anaconda
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev
// 基于Python3.6
pip3 install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

pip安装whl包

torch1.2的包很大，直接用pip安装下载很慢，可以先用下载工具将torch1.2的whl下载下来，再直接安装。

1
pip3 install torch-1.2.0+cu92-cp36-cp36m-win_amd64.whl

验证

打开命令提示符，输入：

1
python

输入如下代码，查看输出：

1
2
3
4
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

输出类似如下结果：

1
2
3
4
5
tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

输入如下代码，验证GPU驱动和CUDA是安装正确，能够被PyTorch访问：

1
2
import torch
torch.cuda.is_available()

参考链接

  1. CUDA,by wikipedia.
  2. Windows 下安装Pytorch,by Big_quant.
  3. Windows系统查看CUDA版本号,by 潇洒坤.
  4. PyTorch,by wikipedia.
  5. GET STARTED,by pytorch.
  6. Getting Started with Python in VS Code,by visualstudio.