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 | // 基于Anaconda |
pip安装whl包
torch1.2的包很大,直接用pip安装下载很慢,可以先用下载工具将torch1.2的whl下载下来,再直接安装。
1 | pip3 install torch-1.2.0+cu92-cp36-cp36m-win_amd64.whl |
验证
打开命令提示符,输入:
1 | python |
输入如下代码,查看输出:
1 | from __future__ import print_function |
输出类似如下结果:
1 | tensor([[0.3380, 0.3845, 0.3217], |
输入如下代码,验证GPU驱动和CUDA是安装正确,能够被PyTorch访问:
1 | import torch |
参考链接
- CUDA,by wikipedia.
- Windows 下安装Pytorch,by Big_quant.
- Windows系统查看CUDA版本号,by 潇洒坤.
- PyTorch,by wikipedia.
- GET STARTED,by pytorch.
- Getting Started with Python in VS Code,by visualstudio.