个人网站设计文字内容模板宁波企业网
背景：我是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这个路径
