update dockerfile

alyashev · Nov 9, 2022 · 90805ed · 90805ed
1 parent 5e7e74a
commit 90805ed
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,2 +1,6 @@
 FROM nvcr.io/nvidia/pytorch:22.09-py3
-RUN pip install wandb tqdm pyyaml
+RUN useradd -ms /bin/bash pino
+USER pino
+ENV PATH=/home/pino/.local/bin:$PATH
+RUN pip install --user \
+    wandb tqdm pyyaml