win10+cuda9+pycharm之pytorch_gpu安装

140 阅读1分钟

不算很难!cuda的安装请自行百度!接下来进入正题!

1.打开Windows命令行

nvcc -V

查看cuda版本信息:

图片.png

2.激活你创造的python虚拟环境,例如我这里是tf_gpu_cuda9:

conda activate tf_gpu_cuda9  #虚拟环境名称

图片.png

3.下载pytorch(GPU版),我已经在网上查过了,我自己的cuda是9.0,cudnn是7.0,可以不用清华源,直接用官网就行。

pip install <http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl>
pip install --no-deps torchvision==0.4.0  #经测试,这里添加的参数不会删除之前下载的torch

4.在python编译器上进行检验,我这里用的是pycharm:

import torch
print(torch.cuda.is_available())

图片.png