配环境时,执行:mim install mmcv==2.0.0rc4 昨天还可以安装成功,今天突然安装不上了
报错如图所示,试了很多方法都不行,还好在组会前解决了这个bug


1. 检查当前 CUDA 版本:
echo $TORCH_CUDA_ARCH_LIST
版本包括:5.2 6.0 6.1 7.0 7.5 8.0 8.6 9.0+PTX
2. 设置环境变量
如果发现不支持 9.0+PTX,可以通过设置环境变量来解决。
export TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6+PTX"
1912

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



