首先,您需要安装tensorflow-gpu ,因为此软件包负责gpu计算。还记得运行环境变量CUDA_VISIBLE_DEVICES = 0来运行代码(或者,如果您有多个GPU,则用逗号将它们的索引放入)。可能与使用gpu有关。如果您的张量流仍然不使用gpu,请尝试此操作

如何使用TensorFlow GPU?

共 2 个回答
高赞
时间
活跃
0

0

遵循本教程Tensorflow GPU,我做到了,它运行完美。
注意! -安装9.0版! Tensorflow-gpu不支持较新的版本
脚步:
- 卸载旧的tensorflow
- 安装tensorflow-gpu
pip install tensorflow-gpu
- 安装Nvidia显卡和驱动程序(您可能已经拥有)
- 下载并安装CUDA
- 下载并安装cuDNN
- 通过简单程序验证
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
新手导航
- 社区规范
- 提出问题
- 进行投票
- 个人资料
- 优化问题
- 回答问题
0
如何在Python 3.6 x64中使用TensorFlow GPU版本而不是CPU版本?
Python正在使用我的CPU进行计算。
我可以注意到它,因为我有一个错误:
我已经安装了tensorflow和tensorflow-gpu。
如何切换到GPU版本?