【发布时间】:2020-10-27 12:51:19
【问题描述】:
我在 Windows 10 中安装了 CPU 和 GPU 版本的 tensorflow。
conda list t.*flow
# packages in environment at C:\Users\Dell\anaconda4:
#
# Name Version Build Channel
tensorflow 2.3.1 pypi_0 pypi
tensorflow-estimator 2.3.0 pypi_0 pypi
tensorflow-gpu 2.3.1 pypi_0 pypi
tensorflow-gpu-estimator 2.3.0 pypi_0 pypi
另外,我已经按照此链接https://towardsdatascience.com/installing-tensorflow-with-cuda-cudnn-and-gpu-support-on-windows-10-60693e46e781 上的步骤安装了 CUDA 和 cuDNN,唯一的区别是我下载了最新版本的 CUDA 和 cuDNN 以符合 tensorflow 2.3.1 的要求,但我仍然可以不访问我的 GPU,这是一个 NVIDIA GeForce MX150。
import tensorflow as tf
tf.test.is_built_with_cuda()
返回真。
tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)
输出:
警告:tensorflow:From :1: is_gpu_available(来自 tensorflow.python.framework.test_util)已弃用,将在未来版本中删除。
更新说明:
请改用tf.config.list_physical_devices('GPU')。
错误
关于为什么 tensorflow 2.3.1 无法访问/找到 GPU 有什么想法吗?请帮我解决这个问题。
【问题讨论】:
-
运行示例程序时,Tensorflow 的完整日志输出是什么?
标签: python tensorflow