From eb60d8529b308b08cd92280b2be3e4677aeca1e1 Mon Sep 17 00:00:00 2001 From: Michael Hansen Date: Fri, 11 Nov 2022 13:26:12 -0500 Subject: [PATCH] Add Docker scripts --- .gitignore | 1 + src/python/.dockerignore | 1 + src/python/Dockerfile | 6 ++++++ src/python/run-docker | 14 ++++++++++++++ 4 files changed, 22 insertions(+) create mode 100644 src/python/.dockerignore create mode 100644 src/python/Dockerfile create mode 100755 src/python/run-docker 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 \ + "$@"