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版本的安装就完成了。
第一次创作不足之处,请多多包涵。

  • 7
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 7
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值