Tensorflow GPU не работает в Pycharm

Когда я запускаю свой обучающий модуль tenorflow в Pycharm IDE в Ubuntu 16.04, он не показывает никаких тренировок с GPU и обычно тренируется с CPU. Но когда я запускаю тот же скрипт на Python с использованием терминала, он запускается с использованием обучения GPU. Я хочу знать, как настроить обучение GPU в Pycharm IDE.

Источник

2 ответа

Решение

На самом деле проблема была в том, что среда python для проекта pycharm отличается от среды, используемой в конфигурациях запуска. Эта проблема была исправлена ​​путем изменения среды в конфигурациях запуска.

Источник

Следующие 4 шага должны решить вашу проблему.

  1. Открыть терминал
  2. активируйте свою виртуальную среду conda активируйте your_virtual_env
  3. запустить Pycharm из терминала pycharm-community
  4. Вам необходимо установить переменную среды для консоли Python [для проекта].

  1. Теперь запустите программу из pycharm, она должна работать с графическим процессором.
Источник
Другие вопросы по тегам