美文网首页
Ubuntu用Anaconda 安装PyTorch、torchv

Ubuntu用Anaconda 安装PyTorch、torchv

作者: 糊糊小鱼干 | 来源:发表于2018-06-02 15:14 被阅读0次

    适用于服务器上多用户安装于自己目录的情况

    Anaconda

    安装Anaconda见此篇

    CUDA

    安装CUDA见此篇

    查看CUDA版本

    nvcc -V
    

    Pytorch & torchvision

    修改软件源为清华的源

    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 
    conda config --set show_channel_urls yes 
    

    去Pytorch官网根据需要的版本下载安装,需要关注的有:

    • OS
    • Package Manager
    • Python
    • CUDA

    本次选择如图

    1

    按照command提示下载

    conda install pytorch torchvision cuda90 -c pytorch
    

    完成之后可在python中查看

    import torch
    import torchvision
    print(torch.__version__)#输出PyTorch版本
    

    相关文章

      网友评论

          本文标题:Ubuntu用Anaconda 安装PyTorch、torchv

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