@安装paddlepaddle-gpu时ImportError: libcudart.so.11.0: cannot open shared object file: No such file or directory

ImportError: libcudart.so.11.0: cannot open shared object file: No such file or directory

在这里插入图片描述
根据报错的提示可以看出应该是CUDA的问题,首先检查服务器GPU对应的CUDA版本nvidia-smi
在这里插入图片描述可以看到本机的CUDA版本为11.0
由此可以看出,上面报错提示的应该是缺少CUDA11.0的相关文件,因此我们安装CUDA11.0

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
sudo sh cuda_11.0.2_450.51.05_linux.run

直接按照提示全部默认安装,但是安装完成之后再执行import paddle还是出现上述的错误,因为我们还没有配置环境变量

export LD_LIBRARY_PATH=/usr/local/cuda-11.X/lib64/:$LD_LIBRARY_PATH

通过上述命令配置环境变量之后再执行import就成功啦
参考
https://blog.csdn.net/qq_38308388/article/details/129528132

Logo

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

更多推荐