解决：Conda虚拟环境中未设置CUDA_HOME的问题

背景：我是Ubuntu22.04系统，最近在复现FoundationPose算法，按照README构建部署环境时，有一步一直卡住，看了下是未找到CUDA_HOME这个环境变量。

网上搜了下这个错误，需要设置CUDA_HOME的环境变量路径，一般是在/usr/local/cuda路径下。

问题

• 我的CUDA和CUDNN是通过conda虚拟环境里面安装的，虽然通过指令可以获取到CUDA和CUDNN的版本，但/usr/local/目录下没有类似cuda的文件夹

python
import torch
print("CUDA Version:", torch.version.cuda)
print("cuDNN Version:", torch.backends.cudnn.version())

• 看到有博文说可以通过指令找到路径，就是在新建的conda环境下

(foundationpose) zzw@zzw:/usr/local$ conda list cudatoolkit
# packages in environment at /home/zzw/anaconda3/envs/foundationpose:
#
# Name                    Version                   Build  Channel
cudatoolkit               8.0                           3    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free

• 于是我输入了CUDA_HOME的环境变量路径

export CUDA_HOME=/home/zzw/anaconda3/envs/foundationpose

• 执行foundationpose的编译程序还是报错了，仍然是找不到相关CUDA的配置

解决

• 尝试了很多办法后，还是无法解决这个问题，索性就在Linux环境下安装一个CUDA和CUDNN（非conda环境）

• 安装CUDA和CUDNN参考：https://blog.csdn.net/qq_45445740/article/details/145809660?spm=1001.2014.3001.5501

• 安装完成后，可以看到有了/usr/local/cuda这个路径（这里因为FoundationPose环境配置里面写的是cuda11.8，为了保持一致避免其他bug，我也安装的11.8CUDA）

(foundationpose) zzw@zzw:/usr/local$ ll
total 56
drwxr-xr-x 14 root root 4096 Feb 23 10:59 ./
drwxr-xr-x 14 root root 4096 Feb 21  2024 ../
drwxr-xr-x  2 root root 4096 Feb 21  2024 bin/
lrwxrwxrwx  1 root root   21 Feb 23 10:59 cuda -> /usr/local/cuda-11.8//
drwxr-xr-x 17 root root 4096 Feb 23 11:00 cuda-11.8/
drwxr-xr-x  2 root root 4096 Feb 21  2024 etc/
drwxr-xr-x  2 root root 4096 Feb 21  2024 games/
drwxr-xr-x  2 root root 4096 Feb 21  2024 include/
drwxr-xr-x  3 root root 4096 Feb 21  2024 lib/
lrwxrwxrwx  1 root root    9 Sep 25 00:21 man -> share/man/
drwx------  3 root root 4096 Jan 15 11:22 nginx/
drwxr-xr-x  3 root root 4096 Dec 17 13:20 Qt-5.6.3/
drwxr-xr-x  2 root root 4096 Feb 21  2024 sbin/
drwxr-xr-x  7 root root 4096 Feb 21  2024 share/
drwxr-xr-x  2 root root 4096 Feb 21  2024 src/
drwxrwxr-x  7 zzw  zzw  4096 Feb 13 16:53 sunlogin/

• 设置CUDA_HOME环境变量，再次编译原先的算法，发现可以编译通过了

export CUDA_HOME=/usr/local/cuda

问题猜测

• 猜测可能是11.8并没有通过conda安装，官方显示不在可用，所以导致conda虚拟环境里面没有/usr/local/cuda这个路径