一、安装Anaconda3
Anaconda下载地址:https://repo.continuum.io/archive/,但由于下载速度极慢,此处分享网盘链接 Anaconda3-4.2.0-Windows-x86_64 ,密码w2qi。(对应python3.5版本)
二、下载安装CUDA 9.0和Cudnn
过程在前面安装keras时已讲过,不在赘述
三、安装pytorch
提前下载好pytorch安装包(windows,python3.6,cuda9.0)
http://download.pytorch.org/whl/cu90/torch-1.0.1-cp36-cp36m-win_amd64.whl
https://pypi.org/project/torchvision/0.2.1/#files
(注意:torch-1.0对应torchvision-0.2,torch-1.1对应torchvision-0.3)
其他版本对应的pytorch包请见官网 https://pytorch.org/ 选pip那项,https那个就是链接了。
* 新建conda环境:conda create -n torch python=3.6
* 激活环境 :conda activate torc