当前位置:   article > 正文

九天·毕昇中国移动平台安装MMdetection_九天毕昇cuda版本

九天毕昇cuda版本

安装MMDetection:

中国移动九天·毕昇平台内置的pytorch1.8环境的pytorch,cuda版本

import torch,torchvision
print(torch.__version__,torch.cuda.is_available())
!nvidia-smi
  • 1
  • 2
  • 3

pytorch:1.8.0,CUDA Version:11.0

问题描述

使用pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.8.0/index.html安装mmcv,会出现以下报错:Failed to build mmcv-ful

原因分析:

查看whl对应版本
可以发现没有对应于pytorch1.8与cuda11.0的版本

解决方案:

可以更改命令为pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html

有问题欢迎+:xzming0908 讨论

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/盐析白兔/article/detail/403112
推荐阅读
相关标签
  

闽ICP备14008679号