问题原因

猜测是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

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐