解决libtorch安装编译链接时出错

最新推荐文章于 2025-09-18 13:15:01 发布

原创最新推荐文章于 2025-09-18 13:15:01 发布 · 2.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #python

图像分割专栏收录该内容

5 篇文章

订阅专栏

该博客讲述了在使用CMake构建过程中遇到的链接错误，涉及torch库依赖于CUDA的函数，如cudaStreamUpdateCaptureDependencies等，解决方法是通过conda安装特定的cudatoolkit-dev补丁包。

部署运行你感兴趣的模型镜像

cmake --build . --config Release
Scanning dependencies of target example-app
[ 50%] Building CXX object CMakeFiles/example-app.dir/example-app.cpp.o
[100%] Linking CXX executable example-app
/usr/bin/ld: /mnt/d70e1bea-6050-40c1-80c9-c5274907367e/anaconda_envs/torch_t/lib/python3.7/site-packages/torch/lib/libtorch_cuda_cpp.so: undefined reference to cudaStreamUpdateCaptureDependencies@libcudart.so.11.0' /usr/bin/ld: /mnt/d70e1bea-6050-40c1-80c9-c5274907367e/anaconda_envs/torch_t/lib/python3.7/site-packages/torch/lib/libtorch_cuda_cpp.so: undefined reference to cudaStreamGetCaptureInfo_v2@libcudart.so.11.0’
/usr/bin/ld: /mnt/d70e1bea-6050-40c1-80c9-c5274907367e/anaconda_envs/torch_t/lib/python3.7/site-packages/torch/lib/libtorch_cuda_cu.so: undefined reference to cudaFreeAsync@libcudart.so.11.0' /usr/bin/ld: /mnt/d70e1bea-6050-40c1-80c9-c5274907367e/anaconda_envs/torch_t/lib/python3.7/site-packages/torch/lib/libtorch_cuda_cu.so: undefined reference to cudaMallocAsync@libcudart.so.11.0’
/usr/bin/ld: /mnt/d70e1bea-6050-40c1-80c9-c5274907367e/anaconda_envs/torch_t/lib/python3.7/site-packages/torch/lib/libtorch_cuda_cpp.so: undefined reference to cudaGraphRetainUserObject@libcudart.so.11.0' /usr/bin/ld: /mnt/d70e1bea-6050-40c1-80c9-c5274907367e/anaconda_envs/torch_t/lib/python3.7/site-packages/torch/lib/libtorch_cuda_cpp.so: undefined reference to cudaUserObjectCreate@libcudart.so.11.0’
collect2: error: ld returned 1 exit status
make[2]: *** [CMakeFiles/example-app.dir/build.make:99：example-app] 错误 1
make[1]: *** [CMakeFiles/Makefile2:76：CMakeFiles/example-app.dir/all] 错误 2
make: *** [Makefile:84：all] 错误 2

解决：conda install -c conda-forge cudatoolkit-dev

您可能感兴趣的与本文相关的镜像