九天·毕昇中国移动平台安装MMdetection

安装MMDetection:

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

import torch,torchvision
print(torch.__version__,torch.cuda.is_available())
!nvidia-smi

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 讨论

你可能感兴趣的:(pytorch,深度学习,人工智能)