Add Docker scripts

pull/2/head
Michael Hansen 2 years ago
parent a6b2d2e69c
commit eb60d8529b

1
.gitignore vendored

@ -14,3 +14,4 @@ htmlcov
*.so
.venv/
lightning_logs/

@ -0,0 +1,6 @@
FROM nvcr.io/nvidia/pytorch:22.03-py3
RUN pip3 install \
'pytorch-lightning'
ENV NUMBA_CACHE_DIR=.numba_cache

@ -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 \
"$@"
Loading…
Cancel
Save