不算很难!cuda的安装请自行百度!接下来进入正题!
1.打开Windows命令行
nvcc -V
查看cuda版本信息:
2.激活你创造的python虚拟环境,例如我这里是tf_gpu_cuda9:
conda activate tf_gpu_cuda9 #虚拟环境名称
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())