Tensorflow python, запуск внутри докера из PyCharm

Хочу поучится Tensorflow. На оффсайте написано, что наиболее правильный вариант для работы TF для GPU - скачать образ для докера, у него уже есть все зависимости и всё хорошо.

При этом в туториале использование докера выглядит примерно так:

docker run --gpus all -it --rm tensorflow/tensorflow:latest-gpu \
   python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

Т.е. код прям отдавать докеру внутри консоли. А как быть, если я хочу запустить проект, состоящий из нескольких классов, да и просто поработать в уютной IDE-шке, например PyCharm? У него есть удобная кнопка "Запуск" сверху, и если у меня локально стоит питон + драйвера на Nvidia + Cudnn - можно выбрать нужный интерпретатор питона в настройках PyCharm и кнопкой "Запуск" в IDE сразу же запускать весь проект через python+tensorflow gpu.

Но как реализовать это, если у меня питон и tf находятся в докере?