-
Notifications
You must be signed in to change notification settings - Fork 2
/
docker-compose.yaml
25 lines (25 loc) · 1.04 KB
/
docker-compose.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
version: "3.8"
services:
ddl:
container_name: docker-deep-learning-debug
# The following lines define the NVIDIA runtime and give the container
# access to all available GPU devices on the host.
# INFO: comment them out if you don't have access to a GPU, as the container will not start otherwise.
runtime: nvidia
environment:
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=all
# The context used when building the image.
# In this case it is the project's root, where the docker-compose build command is executed from
build:
context: ./
#
stdin_open: true
tty: true
# Here we mount the ./src directory containing our code from the host machine into the container.
# This way, our container always gets the most up-to-date state of our code base, without needing to rebuild the image! :)
volumes:
- ./src/:/work
# This forwards the container port 5678 to host port 5678, where we wil start the debugger from within VSCode.
ports:
- 5678:5678