美文网首页深度学习
Ubuntu 16.04下Anoconda安装Pytorch

Ubuntu 16.04下Anoconda安装Pytorch

作者: 逆风g | 来源:发表于2018-11-22 17:38 被阅读33次

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

    相关文章

      网友评论

        本文标题:Ubuntu 16.04下Anoconda安装Pytorch

        本文链接:https://www.haomeiwen.com/subject/czalqqtx.html