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.do