Tensorflow-gpu运行问题

Question

通过以下方式：

• Tensorflow-gpu 2.0.0
• Windows 10环境
• NVIDIA GTX 1050 gpu
• cuda 10.0和相应的cudnn 7.6.5

我遵循了TF-gpu的官方TF文档，并尝试创建并安装了一个简单的CNN模型(在a.py文件中，我尝试过使用jupyter，但是内核总是死掉)，但是我得到了以下内容：

2020-02-06 23:57:14.420911: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_100.dll
2020-02-06 23:57:16.081396: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library nvcuda.dll
2020-02-06 23:57:16.861094: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1618] Found device 0 with properties: 
name: GeForce GTX 1050 major: 6 minor: 1 memoryClockRate(GHz): 1.493
pciBusID: 0000:01:00.0
2020-02-06 23:57:16.861492: I tensorflow/stream_executor/platform/default/dlopen_checker_stub.cc:25] GPU libraries are statically linked, skip dlopen check.
2020-02-06 23:57:16.862290: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1746] Adding visible gpu devices: 0
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
2020-02-06 23:58:14.322053: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
2020-02-06 23:58:14.324900: F tensorflow/stream_executor/lib/statusor.cc:34] Attempting to fetch value instead of handling error Internal: failed to get device attribute 13 for device 0: CUDA_ERROR_UNKNOWN: unknown error

有人知道怎么做才能使TF-GPU2.0.0正常运行吗？我已经用2.1.0进行了测试，但问题似乎仍然存在。

Answer 1

在这里提供解决方案，即使它存在于Github中，也是为了Stackoverflow社区的利益。

使用下面的组合安装Tensorflow_GPU : 2.1.0已经解决了CUDA_ERROR_UNKNOWN问题。

• Python版本: 3.7.6
• 编译器: MSVC 2017
• 库达: 10.1
• cuDNN : 7.6.5

请参考windows、CPU和GPU的经过测试的构建配置。

Answer 2

在我的例子中，限制GPU内存起作用了，正如前面提到的这里。

将以下内容准备到代码中：

import tensorflow as tf

gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(gpus[0], True)