近期很多小伙伴询问我如何安装GPU版本tensorflow,其实特别简单。首先你的电脑一定要有NVIDIA显卡,算力现在的电脑应该普遍满足,不做考虑。
看到网上有许多GPU安装的方法,倍感麻烦,在此做简单安装介绍。
第一步:下载cuda10.0和cudnn
点我获取,提取码:e8fu
cuda10.0安装其实直接默认即可,当然自定义也是可以的。去掉其中的flash插件,其实无伤大雅,方便考虑,默认即可。
cudnn解压后,复制到cuda底下,和bin相同目录即可。
这样就完成了重要的一步。
第二步 : 配置环境变量
如图所示配置即可。
第三步:tensorflow—gpu的安装
pip install tensorflow-gpu==2.0.0
conda install tensorflow-gpu=2.0.0
pip和conda选其一安装即可。当然配置国内镜像会快速很多。
如此tensorflow-GPU版本的安装就完成了。
第一次创作不足之处,请多多包涵。