Работает ли графический процессор google colab с tenorflow 1.0.0?

Question

когда я проверил версию tenorflow в google colab notebook, он был 1.8.0, который уже предустановлен.Я включил графический процессор из настроек ноутбука и проверил его с помощью этой команды

import tensorflow as tf
device_name = tf.test.gpu_device_name()
if device_name != '/device:GPU:0':
  raise SystemError('GPU device not found')
print('Found GPU at: {}'.format(device_name))

, он выдал желаемый результат

, но мне нужен тензор потока 1.0 для моего кода, поэтому я установил его, используя !pip install tensorflow==1.0установлен успешно, но когда я перезапустил среду выполнения и проверил наличие графического процессора, он выдал следующую ошибку

SystemError: Устройство GPU не найдено

Я что-то пропустил или это с colab

Answer 1

Вам нужно изменить версию cuda на 8 ....

это удалит текущую версию:

      !apt-get remove cuda
      !apt-get autoremove cuda
      !apt-get purge cuda
      !apt-key del /var/cuda-repo-9-2-local/*.pub
      !rm -rf /var/cuda-repo-8-0-local-ga2/

А теперь установите версию 8:

     !sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb
     !sudo apt-get update
     !sudo apt-get -y install cuda-8-0

А теперь установите tenorflow gpu:

     !pip install tensorflow-gpu==1.0

, убедитесь, что у вас установлена ​​версия 1.0.

проверьте, работает ли:

    import tensorflow as tf
    tf.test.gpu_device_name()

Answer 2

Это бы сработало, учитывая, что это просто контейнер nvidia-docker (или, возможно, что-то эквивалентное).

Только для того, чтобы вам пришлось вручную устанавливать совместимые CUDA, cuDNN и другие пакеты.Это может быть беспокойный процесс, и я лично не пробовал это.Также вы устанавливаете пакет tenorflow, для которого не включена поддержка gpu.Попробуйте эту же команду, но с tenorflow-gpu, т.е.

!pip install tensorflow-gpu=1.0