1. 程式人生 > >【PyTorch】windows10 + CUDA 9.1 + cuDNN v7.0.5 + PyTorch 0.4.1

【PyTorch】windows10 + CUDA 9.1 + cuDNN v7.0.5 + PyTorch 0.4.1

準備工作:

注意:PyTorch 在 windows 環境下,只支援 Python3。

CUDA:

CUDA:配置環境變數

CUDNN

## 解壓 cudnn-9.1-windows10-x64-v7.zip
## 將資料夾裡內容拷貝至安裝 CUDA 的地方覆蓋對應資料夾,預設資料夾在: 
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1

PyTorch:

pip install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip install torchvision

torch.cuda.is_available() =  false   ???

import torch
torch.cuda.is_available()

## 如果得到的是 false,不妨重啟電腦試試看
## 哈哈,我一開始 torch.cuda.is_available() 就是等於 false
## 然後去下載了 VS2015 打算進行編譯,但又有點小問題編譯不了
## 重啟電腦後,發現 torch.cuda.is_available() = true 了