From 1a51fc6e2b44cfaeffde0991e0b266f54a9939ff Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Tue, 15 Nov 2022 17:19:21 -0800 Subject: [PATCH] Revert "Add temporary fix for CUDA issue in Dockerfile (#5421)" (#5431) (#5432) This reverts commit 0718b17a8e1f89ee7e167698d1a5d5acad3f1b2a. Co-authored-by: yaoyu-33 <54727607+yaoyu-33@users.noreply.github.com> Signed-off-by: andrusenkoau --- Dockerfile | 4 ---- 1 file changed, 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index 5bae884d48da..1d4a2966d7fa 100644 --- a/Dockerfile +++ b/Dockerfile @@ -92,9 +92,5 @@ COPY tests /workspace/nemo/tests COPY tutorials /workspace/nemo/tutorials # COPY README.rst LICENSE /workspace/nemo/ -# Temporary fix CUDA issue -RUN sed -i "s/, all_gpu_ids//g" /opt/conda/lib/python3.8/site-packages/pytorch_lightning/accelerators/cuda.py -RUN sed -i "s/all_gpu_ids =/\# all_gpu_ids =/g" /opt/conda/lib/python3.8/site-packages/pytorch_lightning/accelerators/cuda.py - RUN printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh && \ chmod +x start-jupyter.sh