深度学习

Ubuntu 16.04下Anoconda安装Pytorch

2018-11-22  本文已影响33人  逆风g

目前我已经安装好了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 torchimport 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

上一篇 下一篇

猜你喜欢

热点阅读