Не удается загрузить CUDA. Tensorflow пытается загрузить неправильную версию cudart64

#tensorflow #gpu

#tensorflow #gpu

Вопрос:

Я не могу заставить TensorFlow распознать версию CUDA 11.1 на моем компьютере.

Я обновил файлы пути в новое местоположение, но всякий раз, когда я запускаю обучение, я вижу, что он ищет старую версию ‘cudart64_101.dll — этого больше не существует в моей файловой системе.

 'Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found'
 

Текущая версия библиотеки DLL cudart64 установлена и в моем файле path cudart64_110.dll , а не тот cudart64_101.dll то, что он ищет.

У меня есть компьютер с Windows, и я проверил установку CUDA: nvidia-smi

 Sun Nov 22 12:11:15 2020
 ----------------------------------------------------------------------------- 
| NVIDIA-SMI 456.81       Driver Version: 456.81       CUDA Version: 11.1     |
|------------------------------- ---------------------- ---------------------- 

nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:54:10_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.relgpu_drvr455TC455_06.29190527_0
 

Как мне заставить tensorflow искать правильную cudart64_110.dll файл?

Комментарии:

1. Вы не можете. У вас должна быть установлена версия библиотек времени выполнения CUDA, для которых был создан Tensorflow. Если вы хотите использовать CUDA 11, то вам нужна версия Tensorflow, которая была создана на основе CUDA 11

2. Большое спасибо. Я понизил рейтинг CUDA до версии, совместимой с моей сборкой tensorflow, и это здорово сработало.