一,问题:
安装好paddlepaddle-gpu后使用以下代码测试GPU是否可用:
import paddle
paddle.fluid.install_check.run_check()
但是报错:
二,问题出现原因:
libcudnn和libcublas的软连接没有添加到shared library中
三,解决办法:
1,查找libcudnn.so和libcublas.so的位置:
分别执行sudo find / -name libcudnn.so*
和sudo find / -name libcublas.so*
,如下图:
2,在/usr/lib中建立/usr/lib/x86_64-linux-gnu/libcudnn.so.7.5.0和/usr/local/cuda-10.0/targets/x86_64-linux/lib/libcublas.so.10.0.130的软链接(注意,你查找到的这两个文件很大可能与我不一样):
$ sudo ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.7.5.0 /usr/lib/libcudnn.so
$ sudo ln -s /usr/local/cuda-10.0/targets/x86_64-linux/lib/libcublas.so.10.0.13 /usr/lib/libcublas.so
3,查看/usr/lib下是否已有libcudnn.so和libcublas.so文件:
$ ls /usr/lib/ |grep libcu
如下图即添加成功
完成之后重新运行测试程序,成功:
Done!