【python报错】libcusparse.so.12: undefined symbol: __nvJitLinkAddData_12_1, version libnvJitLink.so.12
·
问题原因
猜测是pytorch的依赖问题,缺少相关库导致gpu加载失败。使用conda install pytorch pytorch-cuda=12.1 -c pytorch -c nvidia也无法修复。
解决方法
ln -s /home/dell/Coding/Envs/miniconda/envs/only_test/lib/python3.10/site-packages/nvidia/nvjitlink/lib/libnvJitLink.so.12 /home/dell/Coding/Envs/miniconda/envs/only_test/lib/python3.10/site-packages/nvidia/cusparse/lib/libnvJitLink.so.12
export LD_LIBRARY_PATH=/home/dell/Coding/Envs/miniconda/envs/only_test/lib/python3.10/site-packages/nvidia/cusparse/lib:$LD_LIBRARY_PATH
需要修改环境路径
参考:https://blog.csdn.net/qq_42730750/article/details/139582293
更多推荐
所有评论(0)