在Windows环境下安装PyTorch
注:这是从官网的快捷安装下总结来的,主要是方便自己查看。
- 安装anaconda,参照链接Anaconda入门
- 按照路径Start | Anaconda3 | Anaconda Prompt打开windows下的终端框。
- 将pytorch安装在对应的虚拟环境中便于管理,使用如下命令创建虚拟环境并进入环境。也可以使用Anaconda软件的图形化界面来创建虚拟环境。
conda create -n torch python=3.5#建立一个名为torch的使用3.5版本python的虚拟环境
conda activate torch
- 进入PyTorch官网找到这个界面。选择想安装的环境和版本,将最下面的命令复制到终端中运行。等待安装完成。(官网说较新的cuda版本会比较好)
注:自己的电脑已经安装了cudatoolkit=10.0,这次安装的是历史版本。
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch #用的这条命令
- 验证是否安装成功先输入
python
,再输入如下的代码。观察输出。
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)
- 检查cuda或gpu是否能否正常使用可以使用下面的代码。
import torch
torch.cuda.is_available()
如果第6步返回的结果为Flase,可以到下面第一个链接里面按照自己的电脑显卡下载最新的显卡驱动更新,双击安装即可,然后重新检查是否可用。
英伟达官网下载驱动
此问题的解决方法的参考链接
2020年1月10日更新
遇到ImportError: /lib64/libstdc++.so.6: version `CXXABI_1.3.9’ not found (required by anaconda3/lib/python3.6/site-packages/scipy/sparse/_sparsetools.cpython-36m-x86_64-linux-gnu.so)报错的处理。
如果在import torch出现了如下或者类似的错误:
ImportError: /lib64/libstdc++.so.6: version `CXXABI_1.3.9' not found (required by anaconda3/lib/python3.6/site-packages/scipy/sparse/_sparsetools.cpython-36m-x86_64-linux-gnu.so)
网上也有解决的方法,可参考这篇博客处理。
因为感觉很麻烦,我直接选择了重装。
这个报错的原因是服务器的环境里没有import torch所需要的那么高的环境,所以分析后感觉是安装的torch版本太高了,就降低了安装版本,在pytorch官网主页选择之前的版本按步骤重新进行安装【就是第4步对应的命令不一样】,这个命令在torch官网找,链接入口。
选择安装torch1.0.0版本后没有问题,建议大家也安装1.0.0以下的版本,防止电脑上的一些环境不兼容。