diff --git a/pytorch.dockerfile b/pytorch.dockerfile
index 534e089..4af61a6 100644
--- a/pytorch.dockerfile
+++ b/pytorch.dockerfile
@@ -59,6 +59,8 @@ sudo apt-get clean
 sudo rm -rf /var/lib/apt/lists/*
 EOF
 
+RUN python${PYTHON_VERSION} -m pip install /home/rocm/torch.whl
+
 # Torch Vision
 WORKDIR /home/rocm/
 RUN git clone --depth 1 https://github.com/pytorch/vision.git -b v0.17.2 vision