TensorFlow 1.0 在 Windows 上看不到 GPU(但 Theano 可以)

2024-01-14

我在 Windows 上运行了 Keras 和 Theano 的安装(按照此tutorial https://github.com/philferriere/dlwin)。现在我尝试将后端切换到张量流 https://www.tensorflow.org/install/install_windows效果很好。

我唯一的问题是Tensorflow 未检测到我的 GPU https://stackoverflow.com/questions/38559755/how-to-get-current-available-gpus-in-tensorflow,而 Theano 则相反:

from tensorflow.python.client import device_lib
def get_available_gpus():
    local_device_protos = device_lib.list_local_devices()
    return [x.name for x in local_device_protos if x.device_type == 'GPU']

没有产生任何结果,但是当使用 Theano 后端运行时,它工作得很好:

C:\Programming\Anaconda3\python.exe D:/cnn_classify_cifar10.py 
Using Theano backend.
DEBUG: nvcc STDOUT nvcc warning : The 'compute_20', 'sm_20', and 'sm_21' architectures are deprecated, and may be removed in a future release (Use -Wno-deprecated-gpu-targets to suppress warning).
mod.cu
    Creating library C:/Users/Alex/AppData/Local/Theano/compiledir_Windows-10-10.0.14393-SP0-Intel64_Family_6_Model_60_Stepping_3_GenuineIntel-3.5.2-64/tmpgsy496fe/m91973e5c136ea49268a916ff971b7377.lib and object C:/Users/Alex/AppData/Local/Theano/compiledir_Windows-10-10.0.14393-SP0-Intel64_Family_6_Model_60_Stepping_3_GenuineIntel-3.5.2-64/tmpgsy496fe/m91973e5c136ea49268a916ff971b7377.exp

Using gpu device 0: GeForce GTX 770 (CNMeM is enabled with initial size: 80.0% of memory, cuDNN 5005)

显然缺少一些配置,但我不知道是什么。为了使 Theano 正确运行,我需要一个名为~/.theanorc包含以下内容:

[global]
device = gpu
floatX = float32

[cuda]
root = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

[nvcc]
flags=-LC:C:\Programming\WinPython-64bit-3.5.2.2\python-3.5.2.amd64\libs

也许缺少类似的东西或者也许我需要添加像 Theano 一样的环境变量? https://github.com/philferriere/dlwin#switching-between-cpu-and-gpu-mode。可能相关问题 https://stackoverflow.com/questions/33659047/where-is-the-configure-of-tensorflow-and-how-to-enable-the-gpu-support在 Linux 上(?)。

完整的安装日志(其中包括一个奇怪的异常)可以在这里找到Gist https://gist.github.com/apacha/a595c244f90a27aced56f67f7598d90d.

有什么想法,如何使 GPU 对 Tensorflow 可见?


在同一台计算机上安装 TensorFlow 和 TensorFlow-GPU 目前可能会导致问题。

安装 1.0 版的tensorflow(仅适用于cpu)或tensorflow-gpu(仅适用于gpu)

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

TensorFlow 1.0 在 Windows 上看不到 GPU(但 Theano 可以) 的相关文章

随机推荐