美文网首页Deep Learning
conda install pytorch torchvisio

conda install pytorch torchvisio

作者: yuanthu | 来源:发表于2020-09-18 13:46 被阅读0次

Pytorch 官网https://pytorch.org/

捕获.PNG

选择合适的版本,安装默认是官网海外服务器,添加国内镜像源就可以实现快速的download,依次执行:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda install pytorch torchvision cudatoolkit=10.1

最后一行代码和上面官网不同的是,去掉了-c pytorch,这样才能换用国内清华镜像。然后大家要是用别的版本只要把cudatoolkit=10.1的10.1换掉就行。

清华镜像官网:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

相关文章

网友评论

    本文标题:conda install pytorch torchvisio

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