解決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