tensorflow2.0稳定版GPU简单安装

近期很多小伙伴询问我如何安装GPU版本tensorflow,其实特别简单。首先你的电脑一定要有NVIDIA显卡,算力现在的电脑应该普遍满足,不做考虑。
看到网上有许多GPU安装的方法,倍感麻烦,在此做简单安装介绍。

第一步:下载cuda10.0和cudnn

点我获取,提取码:e8fu
cuda10.0安装其实直接默认即可,当然自定义也是可以的。去掉其中的flash插件,其实无伤大雅,方便考虑,默认即可。
cudnn解压后,复制到cuda底下,和bin相同目录即可。


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

第二步 : 配置环境变量
如图所示配置即可。

编辑path

第三步:tensorflow—gpu的安装

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

pip和conda选其一安装即可。当然配置国内镜像会快速很多。
如此tensorflow-GPU版本的安装就完成了。
第一次创作不足之处,请多多包涵。

原创文章 1 获赞 4 访问量 80

猜你喜欢

转载自blog.csdn.net/qq_44523137/article/details/106112925