tensorflow 2.2 gpu 支持 Windows

Question

我有以下輸出

import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available())
print(tf.config.list_physical_devices('GPU'))
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

2.2.0
False
[]
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 9877499202779451731
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 5902815618689871957
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 9480702146652446076
physical_device_desc: "device: XLA_GPU device"
]

我應該怎么做才能讓 GPU 在 Windows 上工作？

我在路徑中有以下路徑

C:\\Program Files\\NVIDIA GPU 計算工具包\\CUDA\\v11.1\\bin C:\\Program Files\\NVIDIA GPU 計算工具包\\CUDA\\v11.1\\libnvvp C:\\Program Files\\NVIDIA GPU 計算工具包\\CUDA\\v11 .1\\extras\\CUPTI\\lib64 C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.1\\include C:\\cuda\\bin

Answer 1

我相信你需要 cuda v10.1 而不是 11.1。 我發現最簡單的方法是卸載 tensorflow，如果你用 pip 安裝它。 如果您使用的是 Anaconda，那么我們使用 conda 來安裝 tensorflow。 它將安裝版本 2.1.1、cuda 工具包版本 10.1.243 和 cudnn 版本 7.6.5。 注意 conda 只能安裝 tensorflow 到 2.1.1 版本。 如果你想要 tensorflow 2.2 使用 pip install tensorflow ==2.2.0 安裝它。 安裝 2.1 后。 cuda 工具包和 cudnn 適用於 2.2 版。 否則，您必須執行用於安裝 cuda 11.1 的過程，但這次使用 10.1 下載文件。 另一件事。 使用 python3.7 而不是 3.8。 顯然，當您使用 conda 安裝 tensorflow 時，它不適用於 3.8。