#tensorflow #installation #gpu
#tensorflow #установка #графический процессор
Вопрос:
По какой-то причине мой графический процессор не используется tensorflow. Когда я запускаю tf.test.is_built_with_cuda() , он возвращается False . У меня установлен tensorflow-gpu версии 2.3. Также я установил CUDA 10.1 вместе с cuDNN 7.65 и установил правильные переменные среды.
Когда я запустил nvcc -V консоль, она проверяет, что мой CUDA имеет версию 10.1, V10.1.243
Когда я запустил nvidia-smi , он показывает, что моя версия драйвера равна 460.79, а версия CUDA — 11.2
Может быть, мне нужно понизить мой драйвер GPU?
Комментарии:
1. Ваш
tf.test.is_built_with_cuda()результат кажется довольно убедительным, что вы не используете версию TF с поддержкой GPU. Вам не нужно понижать версию вашего драйвера GPU. Вам практически никогда не нужно этого делать.2. Я попытался удалить tensorflow и переустановить tensorflow-gpu пару раз. Как мне это решить?
3. Можете ли вы поделиться журналами, которые отображаются при импорте tensorflow?
4. @ranka47 как мне получить эти журналы?
Ответ №1:
У меня тоже такая же проблема, но когда я установил PyTorch, он правильно определил мой графический процессор, ранее tf 2.1.0 работал нормально.
Возможно, это ошибка.
Я попытаюсь установить tf 2.1 и посмотреть, обнаруживает ли он графический процессор или нет.
ОБНОВЛЕНИЕ 1/7/2021: попытался установить версию tf 2.1.0 через установку conda, но это не сработало.
Tensorflow смог обнаружить мой графический процессор.Я удалил tf из conda установки и установил его через pip . Я думаю, что версия conda не обнаруживает графический процессор.Убедитесь, что устанавливаемая версия tensorflow соответствует требованиям библиотеки CUDA and cudnn .
Моя текущая конфигурация : tensorflow-gpu : 2.3.0
CUDA :7.6.5
cudnn :10.1