解决OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

问题描述:在服务器上安装某个python包的时候报错:OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

解决思路:就是添加cuda路径

解决方法:

1.进入默认的cuda安装目录查看

cd /usr/local
ls

结果没有cuda目录(但是这个环境中是有cuda的,可以正常跑代码,但是没有目录,原因不详)

bin etc  games  include  lib  man  sbin  share  src

2.重装cuda(离线安装)+配置路径参考这两篇博客:

https://blog.csdn.net/m0_56937307/article/details/123722809

https://wenku.csdn.net/answer/de15e9d8253243aa9a74972e13b23b12

2024.7.12修改:加一个参考网址

ubuntu 添加cuda到bashrc环境变量_cuda bashrc-CSDN博客

(第一篇博客的第1、2步有效,第二篇博客的第4步有效。若第一篇博客的路径写法无效,参考第二篇的写法,注意别忘了CUDA_HOME的路径)

配置路径的时候cuda后面别忘了加版本号(以11.1版本为例)(版本号可以在安装完成后的/usr/local/ 的目录下查看),例如:

(base) root@ddad-0:/usr/local# ls
bin  cuda  cuda-11.1  etc  games  include  lib  man  sbin  share  src
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.1

3.使用以下命令使更改生效

source ~/.bashrc

4.重新安装刚才报错的包,解决问题

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值