我如何使用TensorFlow GPU?

我如何使用TensorFlow GPU?,第1张

我如何使用TensorFlow GPU?

遵循本教程Tensorflow GPU,我做到了,它运行完美。

注意! -安装 9.0版! Tensorflow-gpu不支持较新的版本

脚步:

  1. 卸载旧的tensorflow
  2. 安装tensorflow-gpu
    pip install tensorflow-gpu
  3. 安装Nvidia图形卡和驱动程序(您可能已经拥有)
  4. 下载并安装CUDA
  5. 下载并安装cuDNN
  6. 通过简单程序验证

from tensorflow.python.client import device_libprint(device_lib.list_local_devices())



欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5643761.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存