强烈建议安装anaconda之后再来安装这个pytorch，具体怎么安装百度搜索就知道了。
温馨提示，在安装anaconda的时候记得将“添加到环境变量”(安装的时候是英文的)这一选项选上。
下面假设你已经安装好anaconda了：
- 1.第一步下载pytorch的安装包：
链接: https://pan.baidu.com/s/1mh6U01i 密码: 8yty
- 2.打开命令行进入上面所下载的安装包所在的目录，然后输入如下命令：
conda install --offline pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2
- 3.接下来就是查看是否安装成功
进入python命令行交互界面输入如下
ipython Python 3.6.2 |Anaconda custom (64-bit)| (default, Sep 19 2017, 08:03:39) [MSC v.1900 64 bit (AMD64)] Type 'copyright', 'credits' or 'license' for more information IPython 6.1.0 -- An enhanced Interactive Python. Type '?' for help. In [1]: import torch In [2]:
若没报错则表示安装成功
- 4.查看CUDA和cudnn是否能正常工作
In [6]: x = torch.Tensor([666]) In [7]: x_cuda = x.cuda() In [8]: print(x_cuda) 666 [torch.cuda.FloatTensor of size 1 (GPU 0)] In [9]: print(cudnn.is_acceptable(x_cuda)) True
这样表示cuda和cudnn可以正常使用。
如果cudnn报错，则使用如下代码将其关闭
cudnn.enabled = False
如有意合作或学术讨论欢迎私戳联系~
邮箱:marsggbo@foxmail.com
2017-10-2