目前我已经安装好了Anoconda,并且也配置好了cuda和cudnn,在Anoconda中安装了一个TensorFlow,需要再安装一个Pytorch。
下载Pytorch安装包
为什么要下载Pytorch安装包呢,因为官方给出的conda安装方式,下载Pytorch包速度太慢了,可以从以下地址获取conda安装命令,地址https://pytorch.org
安装包直接去清华源那里下载,地址https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
选择好对应pytorch版本、Python版本、cuda版本、cudnn版本。
安装Pytorch
先去Anoconda中创建一个用于安装Pytorch的虚拟环境,进入创建好的虚拟环境,使用pip命令安装已下载好的Pytorch包:
pip install xxx
这里的xxx
是你下载的Pytorch包所在的位置
之后还需要安装Torchvision包
pip install torchvision
都完成安装之后,运行python,输入import torch
和import torchvision
没有报错,则代表安装成功。
安装Visdom
Visdom是Pytorch中的可视化工具,就像TensorFlow中的Tensorboard,安装命令:
pip install visdom
但是目前还有点问题,当我们第一次启动visdom服务时,python -m visdom.server
会提示不能正常使用,会卡在download script,这是无法访问部分国外网站导致的。
这里我提供了已安装好的visdom包,里面已经包含了需要下载的脚本,大家可以把它拷贝到Pytorch虚拟环境下的lib/python2.7/site-packages中,然后执行python -m visdom.server
就不会卡住了。地址https://github.com/gcfrun/Visdom