mmcv与cuda、pytorch版本兼容要求,见mmcv官方文档:https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html#pip 安装部分。
目前网页上默认最新版2.x版本,若要切换旧版,点击页面左下角切换即可。
-
查看自己的cuda和torch版本:
python -c 'import torch;print(torch.__version__);print(torch.version.cuda)' # pytorch 2.0版本需要cuda11.7及以上 -
点击文档链接 选择自己所需版本,拷贝对应的安装命令,进行安装

<

本文介绍mmcv与cuda、pytorch的版本兼容性要求,并提供安装命令。mmcv现已升级至2.x版本,与1.x版本不完全兼容。针对不同CUDA和PyTorch版本提供了详细的安装指导。


2万+

被折叠的 条评论
为什么被折叠?



