PyTorch的安装
1.下载windows版本的conda包
conda包下载地址:https://pan.baidu.com/s/1dF6ayLr#list/path=%2Fpytorch。大家根据需要选择相应的文件下载。
注:文件名的解释:一般为PACKAGENAME-VERSION-PYTHON_VERSION cu80.tar.bz
PACKAGENAME 分为 pytorch 和 pytorch_legacy, 分别为NT内核版本10和6的两类系统进行编译;VERSION代表 pytorch 的版本;而PYTHON则代表python程序的版本,主要分为3.5和3.6。
2.安装conda包
打开命令行提示符,cd到conda包下载的位置。然后运行下面两个命令:
conda install numpy mkl cffi
conda install --offline pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar
3.测试PyTorch
附一段简单测试CUDA与cuDNN是否工作正常的代码:
# CUDA TEST
import torch
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)
# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))
如果能正常输出结果则表明安装成功。
4.import torch出错处理
如果时原来已经安装了anaconda3,在import torch时可能产生如下错误:
Traceback (most recent call last):
File "test.py", line 2, in <module>
import torch
File "C:\Anaconda3\lib\site-packages\torch\__init__.py", line 41, in <module>
from torch._C import *
ImportError: DLL load failed: The specified module could not be found.
这时我们只需要更新一下Anaconda3的python版本至3.6.1及以上即可。更新的方式有以下两种:
conda install -c anaconda python=3.6.1 # 更新至指定版本
conda updata python # 直接更新