解决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 yo

问题描述:在服务器上安装某个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/m0_56937307/article/details/123722809

https://wenku.csdn/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.重新安装刚才报错的包,解决问题

发布者:admin,转转请注明出处:http://www.yc00.com/web/1755498717a5235424.html

相关推荐

发表回复

评论列表(0条)

  • 暂无评论

联系我们

400-800-8888

在线咨询: QQ交谈

邮件:admin@example.com

工作时间:周一至周五,9:30-18:30,节假日休息

关注微信