Почему в моем ноутбуке Colab не используется графический процессор?

Question

Когда я запускаю код на своем ноутбуке colab после выбора графического процессора, я получаю сообщение о том, что «вы подключены к среде выполнения графического процессора, но не используете графический процессор». Теперь я понимаю, что подобные вопросы задавали раньше, но до сих пор не понимаю, почему. Я использую PCA для набора данных в течение сотен итераций для нескольких испытаний. Без графического процессора это занимает примерно столько же времени, сколько и на моем ноутбуке, которое может составлять> 12 часов, что приводит к тайм-ауту на colab. Ограничен ли графический процессор Colab библиотеками машинного обучения, такими как tenorflow? Есть ли способ обойти это, чтобы я мог воспользоваться преимуществами графического процессора для ускорения анализа?

Answer 1

Colab не ограничивается только Tensorflow.

Colab предлагает три типа сред выполнения: стандартная среда выполнения (с ЦП), среда выполнения графического процессора (включая графический процессор) и среда выполнения TPU (которая включает TPU).

«Вы подключены к среде выполнения графического процессора, но не используете графический процессор» означает, что пользователь подключен к среде выполнения графического процессора, но не использует графический процессор, и поэтому время выполнения процессора будет менее затратным. более подходит.

Следовательно, вы должны использовать пакет, который использует графический процессор, например Tensorflow или Jax. Среда выполнения GPU также имеет CPU, и если вы специально не используете пакеты, которые проверяют GPU, он будет бездействовать.