cuda+cudnn+tensorflow-gpu+keras安装及版本对应

cuda+cudnn+tensorflow-gpu+keras安装及版本对应

本文使用的版本是
cuda10.1
cudnn7.6.5
keras 2.2.4
tensorflow-gpu1.15

cuda安装

1.确定电脑是否有GPU,有则可下cuda;
2.查询电脑支持的cuda版本(注:电脑标注的版本非必须对应版本,如我电脑可使用cuda11.1,但我可下载10.1等版本);
3.确定版本,cuda官网下载对应版本;
4.安装时我选择默认目录下载,避免大坑(血泪教训),不慎版本下载错误,也可卸载,在电脑自带的卸载程序中,选择卸载以NVIDIA开头,以版本号结束的软件卸载。

cudnn安装

1.cudnn需要官网注册,按步骤进行即可;
2.官网下载速度尚可,下载结果为压缩包;
3.解压压缩包,将对应的文件夹中的文件复制到cuda安装目录中对应的文件夹下,包括bin、lib等;(忘截图了,论坛其他博主有贴)

tensorflow-gpu安装

1.背景:已下载anaconda,已创建环境;
2.安装:

pip install keras==2.2.4 -i https://pypi.doubanio.com/simple

参考

keras安装

1.背景:已下载anaconda,已创建环境;
2.安装:

pip install keras==2.2.4 -i https://pypi.doubanio.com/simple

参考

其他问题

报错情况1:Could not load dynamic library ‘cudart64_100.dll’; dlerror: cudart64_100.dll not found
参考

血泪教训

确定好版本再安装!!!!!!

你可能感兴趣的:(tensorflow,keras,cuda,cudnn,gpu)