安装chamferdist遇到自己虚拟环境cuda与系统cuda版本不对应
·
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
前言
提示:这里可以添加本文要记录的大概内容:
例如:随着人工智能的不断发展,机器学习这门技术也越来越重要,很多人都开启了学习机器学习,本文就介绍了机器学习的基础内容。
提示:以下是本篇文章正文内容,下面案例可供参考
一、遇到的错误
RuntimeError:
The detected CUDA version (12.5) mismatches the version that was used to compile
PyTorch (11.7). Please make sure to use the same CUDA versions.
这里报错信息很详细,自己的虚拟环境的pytorch-cude是11.7,系统的cuda是12.5,安装chamferdist是冲突。
二、解决
1.查看现在起作用的是哪个
from torch.utils.cpp_extension import CUDA_HOME
print("CUDA_HOME =", CUDA_HOME)
CUDA_HOME = /usr/local/cuda-12.5
可以看出这里是错的路径
在环境中安装cudatoolkit
conda install cudatoolkit=11.7 -c nvidia
安装完设置一下路径
export CUDA_HOME=$CONDA_PREFIX
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
这个时候再看一下起作用的是哪个路径
>>> from torch.utils.cpp_extension import CUDA_HOME
>>> print(CUDA_HOME)
/home/gyy/.conda/envs/puzzlefusionpp
这个时候就对了
但是安装完还是报错
error: [Errno 2] No such file or directory: '/home/gyy/.conda/envs/puzzlefusionpp/bin/nvcc'
这里是缺少nvcc编译工具
conda install -c conda-forge cudatoolkit-dev=11.7
安装一下
再重新安装
pip install chamferdist==1.0.3
Successfully installed chamferdist-1.0.3
成功
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐



所有评论(0)