如何在Python 3.6 x64中使用 TensorFlow GPU 版本而不是 CPU 版本?
import tensorflow as tf
Python正在使用我的 CPU 进行计算。 我可以注意到它,因为我有一个错误:
您的CPU支持该TensorFlow二进制文件未编译为使用的指令:AVX2
我已经安装了tensorflow和tensorflow-gpu。
如何切换到GPU版本?
遵循本教程Tensorflow GPU,我做到了,它运行完美。
注意! -安装 9.0版! Tensorflow-gpu不支持较新的版本
脚步:
pip install tensorflow-gpu
from tensorflow.python.client import device_lib print(device_lib.list_local_devices())