Я установил tensorflow-gpu 1.15.2 на Python 3.7 venv. У меня не установлен пакет tensorflow.
Я установил CUDA 9.0 (так как я использую tenorflow 1.15) и соответствующий cuDNN для CUDA 9.0. Когда я запускаю алгоритм обучения тензорного потока, он использует мой процессор вместо моего графического процессора. Я запустил:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
И он обнаруживает мой графический процессор (1660 Ti), но он говорит, что мне не хватает пакетов:
2020-04-25 22:02:12.536321: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
2020-04-25 22:02:15.175536: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
2020-04-25 22:02:15.188183: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library nvcuda.dll
2020-04-25 22:02:15.234070: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1639] Found device 0 with properties:
name: GeForce GTX 1660 Ti major: 7 minor: 5 memoryClockRate(GHz): 1.875
pciBusID: 0000:27:00.0
2020-04-25 22:02:15.239530: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cudart64_100.dll'; dlerror: cudart64_100.dll not found
2020-04-25 22:02:15.242919: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cublas64_100.dll'; dlerror: cublas64_100.dll not found
2020-04-25 22:02:15.251483: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cufft64_100.dll'; dlerror: cufft64_100.dll not found
2020-04-25 22:02:15.255358: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'curand64_100.dll'; dlerror: curand64_100.dll not found
2020-04-25 22:02:15.266446: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cusolver64_100.dll'; dlerror: cusolver64_100.dll not found
ired libraries for your platform.
Skipping registering GPU devices...
2020-04-25 22:02:15.719511: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1180] Device interconnect StreamExecutor with strength 1 edge matrix:
2020-04-25 22:02:15.721901: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1186] 0
2020-04-25 22:02:15.723610: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1199] 0: N
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 524594082372294943
]
Я выполнил поиск на моем P C и действительно не может найти эти DLL-файлы.
Когда я установил CUDA и попытался переустановить, он говорит: «Графический драйвер не может найти совместимое графическое оборудование». Я проигнорировал ошибку и все равно установил. Соответствующие каталоги CUDA в моих программных файлах находятся там. Но nvcc -V в приглашении cmd подтверждает, что CUDA установлен.
Казалось, быстрый Google предлагает CUDA версии 10+ с этими библиотеками, но после установки я не увидел никакой разницы.
Где я могу найти эти недостающие библиотеки? Я что-то неправильно установил?