解决Ubuntu 18.04下CUDA报错:ImportError: libcudart.so.10.1: cannot open shared object file

1. 问题描述

如题,在Ubuntu 18.04系统下,Python代码调用CUDA报错:

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



2. 解决办法

S1. 在终端中运行命令:

sudo gedit /etc/ld.so.conf.d/nvidia.conf

S2. 在打开的gedit编辑器中输入如下内容(注意里面的路径要按照自己实际情况来写):

/usr/local/cuda-10.1/lib64
/usr/local/cuda-10.1/lib

S3. 保存文本并关闭gedit,然后在终端中运行如下命令:

sudo ldconfig

至此,问题得到解决。



注: 若按照如上步骤操作仍然报错,请将环境变量中其他版本的CUDA注释掉刷新,再测试。



参考资料:

https://stackoverflow.com/questions/36159194/tensorflow-libcudart-so-7-5-cannot-open-shared-object-file-no-such-file-or-di

©️2020 CSDN 皮肤主题: 大白 设计师: CSDN官方博客 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值