Skip to content

Tutorial repo on debugging a PyTorch Neural Network model running inside a Docker container using VSCode.

Notifications You must be signed in to change notification settings

StefanieStoppel/dockerized-deep-learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

37 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

dockerized-deep-learning

🐛️🧠️🐋️ Debugging Deep Learning Docker Containers

This repo goes hand in hand with this Medium post I wrote.

Getting started

Clone this project: git clone [email protected]:StefanieStoppel/dockerized-deep-learning.git.

Prerequisites

You need to have the following things installed on your system:

Create and activate the conda environment

Open a terminal and cd into the directory you cloned the project into. Execute the following:

conda env create --file=environment.yml
conda activate docker-deep-learning

Build the Docker image

In the root directory execute the following:

docker-compose build ddl

Debug the application

  • Open the project in VSCode.
  • Set a breakpoint in a Python file.
  • Open the terminal in the project root and run docker-compose up ddl to run the Docker container.
  • Click on the ▶️play icon in the left hand sidebar of VSCode to launch the debugger.
  • Wait until it stops at you breakpoint.

🐛️ Happy debugging! 🐛️

About

Tutorial repo on debugging a PyTorch Neural Network model running inside a Docker container using VSCode.

Topics

Resources

Stars

Watchers

Forks