问题描述:在服务器上安装某个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.重新安装刚才报错的包,解决问题