【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 了

猜你喜欢

转载自blog.csdn.net/Houchaoqun_XMU/article/details/83043869