From 6c902d6f75023daa893a93c5c0563b8edbaff57b Mon Sep 17 00:00:00 2001
From: Tongze Wang <wtongze@gmail.com>
Date: Thu, 12 Sep 2024 13:06:24 -0700
Subject: [PATCH] fix: torch

---
 pytorch.dockerfile | 2 ++
 1 file changed, 2 insertions(+)

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