Рекомендации по tensorflow 1.13 на cuda 10.1

#tensorflow

#tensorflow

Вопрос:

Я немного поторопился и обновился до Ubuntu 18.10, не смог найти для него версию cuda 10.0, поэтому выбрал cuda 10.1 … которую, как я понимаю, tensorflow пока не поддерживает.

Не посоветовали бы вы вернуться к Ubuntu 18.04 или терпеливо дождаться совместимого выпуска tensorflow?

Ответ №1:

tensorflow 1.13 не работает с cuda 10.1 из-за следующего

Ошибка импорта: libcublas.so.10.0: не удается открыть общий объектный файл: такого файла или каталога нет.

tensorflow ищет libcublas.so.10.0, тогда как cuda предоставляет libcublas.so.10.1.0.105.

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

1. Теперь я перешел на tensorflow 2.0, и он работает нормально.

Ответ №2:

Более старые драйверы cuda доступны здесь: https://developer.nvidia.com/cuda-toolkit-archive

Двоичные файлы, предоставленные TensorFlow, перешли с CUDA 9.0 на CUDA 10.0, пропустив CUDA 9.1 и 9.2. Поэтому я бы не рекомендовал ждать двоичных файлов CUDA 10.1 TensorFlow.

Весьма вероятно, что CUDA 10.0 будет нормально работать на Ubuntu 18.10, но если это не так, то да, вернитесь к версии 18.04.

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

1. Спасибо! Я установил CUDA 10.0, используя предоставленную версию 18.04, и, похоже, на данный момент она работает просто отлично. Я сообщу здесь, если обнаружу какие-либо проблемы.

2. Вы можете собрать TensorFlow 1.14 из исходного кода, используя CUDA 10.1. Или, может быть, Anaconda предоставляет сборку с поддержкой 10.1