参考链接:https://github.com/PaddlePaddle/Paddle/issues/22996
安装Paddle1.8.3之后,出现问题:
解决方案:
(1)对应cuda版本,安装对应的cudnn
旧cudnn版本下载位置:https://developer.nvidia.com/rdp/cudnn-archive
(2)解压之后,将cudnn文件拷贝至cuda10.1(我使用的cuda版本是10.1)中
sudo cp ~/Downloads/cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp ~/Downloads/cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h /usr/local/cuda-10.1/lib64/libcudnn*
如果使用了虚拟链接,可对应修改虚拟链接:
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda
修改/etc/profile文件:
gedit /etc/profile
source /etc/profile
此时,执行paddle.fluid.install_check.run_check(),可显示成功!