oof

sunu · Apr 24, 2024 · 8c827b0 · 8c827b0
1 parent bca8605
commit 8c827b0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,7 +1,6 @@
 FROM nvcr.io/nvidia/pytorch:23.10-py3
 
-# Install the torchrun package.
-RUN pip install torchrun lightning
+RUN pip install lightning
 
 # Copy your training script to use with torchrun.
 COPY ./train.py ./