強烈建議安裝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