在已安装cuda、cudnn基础上,简要记录一下安装流程:
(1)以管理员方式运行Anaconda Prompt,创建一个pytorch环境:
conda create -n pytorch python=3.5 numpy pyyaml mkl
(2)激活pytorch环境:activate pytorch
(3)增加一个镜像源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes //设置是否在安装的时候显示该链接
(4)之前用的pip版本是9.0.1,更新到10.0.1,输入命令:
python -m pip install --upgrade pip
(5)安装pytorch 0.4.0 gpu,python版本是3.5.4,输入命令:
pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl
有可能存在网络问题,尝试多次后终于成功。
(6)安装torchvision:
pip install torchvision
(7)测试:
>>> import torch
>>> a = torch.Tensor([1.])
>>> a.cuda()
tensor([ 1.], device='cuda:0')
>>> from torch.backends import cudnn
>>> cudnn.is_acceptable(a.cuda())
True