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

本文讲述了在使用AdelaiDet库时遇到的安装错误,原因是本地CUDA12.2与PyTorch11.7版本不匹配。作者通过将全局环境的CUDA版本降级到11.4并设置CUDA_HOME解决此问题,同时提到Anaconda环境可能也有CUDA目录影响。
摘要由CSDN通过智能技术生成

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环境的目录,本次就先蹭一下全局环境解决了。

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值