人工智能 python 【Debug日记】The detected CUDA version(12.2) mismatches version that was used to compile PyTorch(11.7)

Bug描述

使用AdelaiDet库，需要本地安装，命令为

pip install -e .

或者

python setup.py install

安装时候出现报错

RuntimeError:

The detected CUDA version (12.2) mismatches the version that was used to compile

PyTorch (11.7). Please make sure to use the same CUDA versions.

分析原因

由于本地CUDA环境已经为12.2，anaconda环境cuda版本为11.7，两者跨越一个大版本容易不兼容，出现安装不成功的情况。

解决问题

一般全局环境CUDA版本升级会有一个过去的备份，找到路径，默认为

/usr/local/cuda-x # x 为版本号

我这里有一个cuda-11.4的环境，抱着试试看的想法export一下完美解决

export CUDA_HOME=/usr/local/cuda-11.4

安装成功~

ps：之前注意到貌似anaconda文件夹里面也会有cuda环境的目录，本次就先蹭一下全局环境解决了。

参考文章

 您阅读本篇文章共花了：