天天看点

tensorflow2.0稳定版GPU简单安装

近期很多小伙伴询问我如何安装GPU版本tensorflow,其实特别简单。首先你的电脑一定要有NVIDIA显卡,算力现在的电脑应该普遍满足,不做考虑。

看到网上有许多GPU安装的方法,倍感麻烦,在此做简单安装介绍。

第一步:下载cuda10.0和cudnn

点我获取,提取码:e8fu

cuda10.0安装其实直接默认即可,当然自定义也是可以的。去掉其中的flash插件,其实无伤大雅,方便考虑,默认即可。

cudnn解压后,复制到cuda底下,和bin相同目录即可。

tensorflow2.0稳定版GPU简单安装

这样就完成了重要的一步。

第二步 : 配置环境变量

如图所示配置即可。

tensorflow2.0稳定版GPU简单安装

第三步:tensorflow—gpu的安装

pip install tensorflow-gpu==2.0.0
           
conda install tensorflow-gpu=2.0.0
           

pip和conda选其一安装即可。当然配置国内镜像会快速很多。

如此tensorflow-GPU版本的安装就完成了。

第一次创作不足之处,请多多包涵。

继续阅读