​​​​​​​​​​​​​​https://mmcv.readthedocs.io/en/latest/get_started/installation.html
上面这是mmcv官网

选择自己对应版本安装即可,一定要安装mmcv-full版本

print("PyTorch version:", torch.__version__)
print("CUDA version:", torch.version.cuda)
这两行是查看自己pytorch以及cuda版本,python版本看编译环境即可
​​​​​​​

完成!收工!

 

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐