欢迎光临
我们一直在努力

TensorFlow GPU的安装和使用

本节将学习如何在 GPU 中使用 TensorFlow,所执行的运算操作是在 CPU 或 GPU 上进行简单的矩阵乘法。

首先安装一个支持 GPU 的 TensorFlow 版本。官方的 TensorFlow 安装说明能够提供良好的开始:https://www.tensorflow.org/。请记住,对 GPU 的环境支持需要通过 CUDA 或 CuDNN。

GPU上TensorFlow的使用

  1. 导入几个模块:



     

赞(0) 打赏
未经允许不得转载:hosteons中文网 » TensorFlow GPU的安装和使用

评论 抢沙发

觉得文章有用就打赏一下文章作者

非常感谢你的打赏,我们将继续提供更多优质内容,让我们一起创建更加美好的网络世界!

支付宝扫一扫

微信扫一扫

登录

找回密码

注册