一尘不染

我如何使用TensorFlow GPU?

python

如何在Python 3.6 x64中使用 TensorFlow GPU 版本而不是 CPU 版本?

import tensorflow as tf

Python正在使用我的 CPU 进行计算。
我可以注意到它,因为我有一个错误:

您的CPU支持该TensorFlow二进制文件未编译为使用的指令:AVX2

我已经安装了tensorflow和tensorflow-gpu。

如何切换到GPU版本?


阅读 154

收藏
2020-12-20

共1个答案

一尘不染

遵循本教程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_lib print(device_lib.list_local_devices())

2020-12-20