Try adding another image layer

stackhpc · Sep 22, 2023 · 4fd0031 · 4fd0031
1 parent 863a420
commit 4fd0031
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/images/pytorch-benchmarks/Dockerfile b/images/pytorch-benchmarks/Dockerfile
@@ -2,6 +2,8 @@ FROM nvcr.io/nvidia/pytorch:22.12-py3
 
 # RUN pip install -U pip
 # RUN pip install torchvision torchaudio
+RUN pip uninstall torch -y && pip install --no-cache torch torchvision torchaudio
+
 # RUN git clone https://github.com/pytorch/benchmark
 # WORKDIR /workspace/benchmark
 # # Add other benchmarks here?