diff --git a/images/pytorch-benchmarks/Dockerfile b/images/pytorch-benchmarks/Dockerfile
index 48e708c..18663eb 100644
--- 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?