我正在使用MSI GL63 8RD-455IN 2018笔记本电脑,具有以下硬件和软件功能:
- NVidia GTX 1050 ti
- Driver version - Nvidia 398.92
- OS : Win 10/64
- keras version : 2.2.2
- python : 3.6.5
- tensorflow version : 1.10.0 - gpu
问题是,tensorflow没有使用GPU。我安装了Cuda 9和Cudnn库,并且仅在成功安装cuda之后才安装tensorflow。
我还进行了以解决方案表示的主要检查,以查看是否正在使用GPU,结果如下:
print(device_lib.list_local_devices())
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 2797024620588530864
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 3168377241
locality {
bus_id: 1
links {
}
}
incarnation: 6850620683865695635
physical_device_desc: "device: 0, name: GeForce GTX 1050 Ti, pci bus id:
0000:01:00.0, compute capability: 6.1"
]
我怀疑的原因是来自 MSI Dragon Center 的GPU使用率指标。尽管可能有人提出这不是适当的度量标准,但我注意到,在用于游戏时,该指示器显示了GPU的使用,但是在运行代码时,似乎仅使用了内置RAM。
如何获取我的代码以使用GPU?
更多信息
此link
中的笔记本电脑信息请告知是否需要更多信息。