Fix up docker

src/python/Dockerfile

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

src/python/run-docker

@@ -1,13 +1,13 @@
 #!/usr/bin/env bash
-nvidia-docker run \
+
+# Follow instructions here: https://docs.docker.com/config/containers/resource_constraints/#access-an-nvidia-gpu
+docker run \
   -it \
+  --gpus all \
   -w "$PWD" \
-  --user $(id -u):$(id -g) \
+  --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 "${HOME}:${HOME}" \
   -v /etc/hostname:/etc/hostname:ro \
   -v /etc/localtime:/etc/localtime:ro \
   larynx2-train \