?

由于pytorch官網點擊打開鏈接提供了windows的安裝方式，故直接采用這種方法，當然也可以通過anaconda來安裝。

由于是安裝gpu的版本，故需要先安裝cuda以及與其版本相對應的cudnn。

本人電腦是華碩，i5，64位，NVIDIA GeForce 940MX，已安裝VS2015。查看gpu屬性的方式：“此電腦”右鍵“屬性”→“設備管理器”→“顯示適配器”，會跳出具體的版本。

滿足cuda所提供的安裝配置需求

“https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#axzz4TpI4c8vf”。

1，安裝顯卡驅動，本機默認已安裝。

2，cuda:

進入“https://developer.nvidia.com/cuda-downloads”，選擇對應的系統配置下載cuda.xxxxxx.exe文件

類似一般性的windows應用的安裝方式安裝即可（目前最新的是cuda 9.2,本人安裝的是9.1）。

驗證是否已安裝：打開cmd,輸入：nvcc??-V，輸出：

表示已安裝。還有其余的驗證步驟可參考“https://blog.csdn.net/fengbingchun/article/details/53892997”

3，cudnn:

官網：https://developer.nvidia.com/cudnn

相比標準的cuda，它在一些常用的神經網絡操作上進行了性能的優化，比如卷積，pooling，歸一化，以及激活層等等。

所以配置cuDNN時是要對cuda進行一些修改，所以我們要先安裝cuda。cuDNN下載需要注冊。

下載之后，

(1)解壓：會生成cuda/include、cuda/lib、cuda/bin三個目錄；

(2)分別將cuda/include、cuda/lib、cuda/bin三個目錄中的內容拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5對應的include、lib、bin目錄下即可。

(3)將bin所在的目錄添加到環境變量 PATH 中，“此電腦”→“高級系統設置”→“環境變量”→“系統變量”→“path”→“編輯”→“新建”加入該路徑即可。

4，pytorch:

本人選擇pip安裝，'pip,cuda9.1,python3.6'，cmd輸入：

?pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl

安裝 torchvision，通過本地安裝的方式進行安裝的，

下載地址：（https://pypi.python.org/pypi/torchvision/0.2.1）

用cmd進入torchvision所在的文件夾下，輸入下面的命令安裝：

pip install torchvision-0.2.1-py2.py3-none-any.whl

5，測試torch能否用gpu:

>>>import torch

>>>torch.__version__

>>>torch.cuda.is_available()

輸出True即可。

此部分可參考“https://blog.csdn.net/sunqiande88/article/details/80085569”
?

總結

以上是生活随笔為你收集整理的Windows10+GPU版 pytorch 安装的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。