import torch
print(torch.version.cuda)
输出的结果中包含CUDA的版本号,可以与mmcv的要求进行比较。
sudo apt-get update
sudo apt-get install gcc g++
pip install --upgrade pip setuptools
pip install git+https://github.com/open-mmlab/mmcv.git
注意:在进行源代码安装时,需要先安装相应的依赖库。具体方法可以参考mmcv官方文档。