Графический процессор Google Colab: []
Я работаю с моделью Maskrcnn в Google Colab. Я выбираю графический процессор в разделе времени выполнения.
import tensorflow as tf
print(tf.__version__)
print(tf.test.gpu_device_name())
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
Выход:
2.18.0-dev20240613
/device:GPU:0
Num GPUs Available: 1
И я проверил, здесь нет проблем.
Когда я установил требования для Maskrcnn, !pip install -r /content/maskrcnn/requirements.txt
2.14.0
Num GPUs Available: 0
ГПУ исчез.
требования.txt
cython==3.0.5
h5py==3.9.0
imgaug==0.4.0
ipython==7.34.0
ipython-genutils==0.2.0
ipython-sql==0.5.0
keras==2.14.0
matplotlib==3.7.1
numpy==1.23.5
opencv-contrib-python==4.8.0.76
opencv-python==4.8.0.76
pillow==9.4.0
scikit-image==0.19.3
scipy==1.11.3
tensorboard==2.14.1
tensorflow==2.14.0
Я не понимаю, почему исчезает графический процессор. У кого-нибудь есть идеи ? Спасибо, С наилучшими пожеланиями.
Привет, спасибо за ответ. Сначала я подключаюсь к графическому процессору, затем устанавливаю Tensorflow. @Джонни Чизкаттер
посмотрите, что ваш requements.txt есть tensorflow==2.14.0 - чтобы он мог удалить ваш tensorflow 2.18 (который работает с графическим процессором) и установить tensorflow 2.14.0, с которым, возможно, есть проблемы. Так что, возможно, установите tensorflow 2.18 еще раз.
Привет, tensorflow==2.15. Я установил эту версию и запустил ее снова, и теперь обучение происходит быстрее. Большое спасибо. @furas, если напишешь это как ответ, я отмечу как правильный. С наилучшими пожеланиями, Большое спасибо,
Привет, tensorflow==2.15. Я установил эту версию и запустил ее снова, и теперь обучение происходит быстрее. Большое спасибо. @Johnny Cheesecutter, если напишешь это как ответ, я отмечу его как правильный. С наилучшими пожеланиями, Большое спасибо,
Ответы 1
Посмотрите, что ваш requements.txt есть tensorflow==2.14.0 - чтобы он мог удалить ваш tensorflow 2.18 (который работает с графическим процессором) и установить tensorflow 2.14.0, у которого могут быть проблемы с графическим процессором.
Поэтому вам придется установить tensorflow 2.18 заново.
Вы пытались перезапустить ядро после установки tensorflow? Или установить тензорный поток перед его импортом?