提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档


前言

提示:这里可以添加本文要记录的大概内容:

例如:随着人工智能的不断发展,机器学习这门技术也越来越重要,很多人都开启了学习机器学习,本文就介绍了机器学习的基础内容。


提示:以下是本篇文章正文内容,下面案例可供参考

一、遇到的错误

      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

成功

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐