【发布时间】:2021-08-28 20:58:03
【问题描述】:
操作系统:Windows 10(也尝试使用 Ubuntu 无济于事)
CPU:英特尔酷睿 i7-8750H @ 2.2 GHz
GPU:采用 Max-Q 设计的 NVIDIA GeForce GTX 1060
Python 3.8 版
TensorFlow 2.3.0 版(我确实使用tensorflow-gpu 安装了它)
代码:
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
print("Num CPUs Available: ", len(tf.config.list_physical_devices('CPU')))
结果:
Num GPUs Available: 0
Num CPUs Available: 1
有什么想法吗?
【问题讨论】:
-
你安装了CUDA吗?
-
如果您正在进行 conda 安装,anaconda 将为您获取 cudnn 和 cuda 工具包依赖项并将其添加到路径中。如果您使用 pip 安装,则必须手动获取所有依赖项。如果您正在使用 tensorflow-gpu,我强烈建议您使用 anaconda。使用方法如下:towardsdatascience.com/…
-
是的,我已经安装了 CUDA
-
您能告诉我们安装的 CUDA 和 cuDNN 的版本吗?
-
我升级到 Python 3.9.6,现在它可以工作了。谢谢
标签: python tensorflow gpu