Conda 离线安装pytorch

Conda 离线安装pytorch

下载安装包,下载对应版本:

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

conda install --offline  pytorch-1.0.0-py3.6_cuda9.0.176_cudnn7.4.1_1.tar.bz2

备注:根据显卡驱动版本号来确定CUDA版本,根据CUDA版本来选择torch,

可根据下文链接选择合适的torch

pytorch-0.4.1-py37_cuda8.0.61_cudnn7.1.2_1.tar.bz2

#它就是适合python3.7.X、cuda8.0、cudnn7.1.2的torch,一般只考虑Python和cuda版本即可,这两个合适一般#cudnn也合适。

参考链接:
该链接给出了显卡信息和对应cuda选择

https://zhuanlan.zhihu.com/p/106133822

posted @ 2020-07-07 13:29  胖墩哥  阅读(4367)  评论(0编辑  收藏  举报