Revert "Add temporary fix for CUDA issue in Dockerfile (NVIDIA#5421)" (…

…NVIDIA#5431) (NVIDIA#5432) This reverts commit 0718b17. Co-authored-by: yaoyu-33 <[email protected]> Signed-off-by: andrusenkoau <[email protected]>
andrusenkoau · Jan 5, 2023 · 1a51fc6 · 1a51fc6
1 parent a4e6da5
commit 1a51fc6
Showing 1 changed file with 0 additions and 4 deletions.
diff --git 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