diff --git a/.gitignore b/.gitignore index 30b9ba8..8e72809 100644 --- a/.gitignore +++ b/.gitignore @@ -14,3 +14,4 @@ htmlcov *.so .venv/ +lightning_logs/ diff --git a/src/python/.dockerignore b/src/python/.dockerignore new file mode 100644 index 0000000..72e8ffc --- /dev/null +++ b/src/python/.dockerignore @@ -0,0 +1 @@ +* diff --git a/src/python/Dockerfile b/src/python/Dockerfile new file mode 100644 index 0000000..cff64f7 --- /dev/null +++ b/src/python/Dockerfile @@ -0,0 +1,6 @@ +FROM nvcr.io/nvidia/pytorch:22.03-py3 + +RUN pip3 install \ + 'pytorch-lightning' + +ENV NUMBA_CACHE_DIR=.numba_cache diff --git a/src/python/run-docker b/src/python/run-docker new file mode 100755 index 0000000..7598f90 --- /dev/null +++ b/src/python/run-docker @@ -0,0 +1,14 @@ +#!/usr/bin/env bash +nvidia-docker run \ + -it \ + -w "$PWD" \ + --user $(id -u):$(id -g) \ + --ipc=host \ + -v "${HOME}/opt:${HOME}/opt" \ + -v /media/data:/media/data:ro \ + -v /media/12tb:/media/12tb:ro \ + -v /media/cache:/media/cache:ro \ + -v /etc/hostname:/etc/hostname:ro \ + -v /etc/localtime:/etc/localtime:ro \ + larynx2-train \ + "$@"