潍坊网站建设价格,ppt模板下载简约,wordpress百度百家模板,wordpress 打开慢背景#xff1a;我是Ubuntu22.04系统#xff0c;最近在复现FoundationPose算法#xff0c;按照README构建部署环境时#xff0c;有一步一直卡住#xff0c;看了下是未找到CUDA_HOME这个环境变量。 网上搜了下这个错误#xff0c;需要设置CUDA_HOME的环境变量路径#x… 背景我是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) zzwzzw:/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?spm1001.2014.3001.5501 安装完成后可以看到有了/usr/local/cuda这个路径这里因为FoundationPose环境配置里面写的是cuda11.8为了保持一致避免其他bug我也安装的11.8CUDA
(foundationpose) zzwzzw:/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这个路径
