cuda9.0-9.2环境安装pytorch>=1.5

举报
qingchenwuhou 发表于 2021/02/02 17:38:33 2021/02/02
【摘要】 cuda9.0-9.2环境安装pytorch1.5,解决torch.cuda.is_available()返回False的问题。

       最近项目用到一阶段检测模型RetinaNet,想用detectron2包来训练模型,detectron2安装环境依赖pytorch>=1.5,我在安装pytorch的时候,可能由于网上下载的编译好的torch和torchvision与本地cuda和cudnn版本不兼容,我在使用pip和conda安装一直出现torch.cuda.is_available()返回False的问题,自己校对了pytorch安装包依赖的环境和本地cuda版本,终于成功安装pytorch。

安装pytorch

方案一:从官网用conda安装,https://pytorch.org/get-started/previous-versions/

方案二:我使用conda安装非常慢,没安装上,所以寻求其他方法:

1.conda create -n  pytorch_1_5_cuda92   python=3.7.6

2.从下载对应torch和torchvision安装包(torch与torchvision版本对应关系从方案一的命令可以查询):

https://download.pytorch.org/whl/torch_stable.html

在服务器上执行(我的本机cuda是cuda=0.9,我网上下载的cu92对应的torch、torchvision安装文件,本地安装也成功):

wget https://download.pytorch.org/whl/cu92/torch-1.5.0%2Bcu92-cp37-cp37m-linux_x86_64.whl

wget https://download.pytorch.org/whl/cu92/torchvision-0.6.0%2Bcu92-cp37-cp37m-linux_x86_64.whl

pip install torch-1.5.0+cu92-cp37-cp37m-linux_x86_64.whl  torchvision-0.6.0+cu92-cp37-cp37m-linux_x86_64.whl

3.python 运行import torch    print(torch.version.cuda)     torch.cuda.is_available()     出现True则安装成功;

4.如果还不成功,可以依靠博客[1]讲的三种方法安装cuda9.0-9.2和cudnn环境到指定的安装目录,然后将指定的安装目录加入环境变量和Lib的搜寻路径。

5. 我在服务器上安装torch.__version__='1.5.0+cu92',对应的cuda与cudnn版本如下:

 export PATH=XXX/lib/cuda-10.0/bin:${PATH}

export LD_LIBRARY_PATH=XXX/lib/cudnn-10.0-linux-x64-v7.5.0.56/lib64:${LD_LIBRARY_PATH}

export LD_LIBRARY_PATH=XXX/lib/cuda-10.0/lib64:${LD_LIBRARY_PATH}

export PATH=XXX/lib/cuda-11.0/bin:${PATH}

export LD_LIBRARY_PATH=XXX/lib/cuda-11.0/lib64:${LD_LIBRARY_PATH}

export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:${LD_LIBRARY_PATH}

Ref:

[1] https://zhuanlan.zhihu.com/p/68069328

【版权声明】本文为华为云社区用户原创内容,未经允许不得转载,如需转载请自行联系原作者进行授权。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。