From 64e492f8deda693def2293d060bb0a7cbddd983e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Irving=20A=2E=20Berm=C3=BAdez=20S?= <bigchirv@gmail.com>
Date: Fri, 12 May 2023 06:39:18 -0400
Subject: [PATCH] Use the torch build whls from evshiron/rocm_lab instead of
 building it from sources

---
 dockerfiles/pytorch2.0_gfx1100.dockerfile | 24 +++++++----------------
 1 file changed, 7 insertions(+), 17 deletions(-)

diff --git a/dockerfiles/pytorch2.0_gfx1100.dockerfile b/dockerfiles/pytorch2.0_gfx1100.dockerfile
index 2a60235..caa550e 100644
--- a/dockerfiles/pytorch2.0_gfx1100.dockerfile
+++ b/dockerfiles/pytorch2.0_gfx1100.dockerfile
@@ -1,24 +1,14 @@
 FROM rocm5.5_ubuntu22.04
 
 ENV HIP_VISIBLE_DEVICES=0 PYTORCH_ROCM_ARCH="gfx1100" USE_CUDA=0
-RUN mkdir -p /srv/src
 RUN pip install cmake ninja mkl mkl-include
+RUN mkdir -p /srv/src
 WORKDIR /srv/src
-RUN wget -nv https://github.com/pytorch/pytorch/releases/download/v2.0.0/pytorch-v2.0.0.tar.gz \
-    && tar zxf pytorch-v2.0.0.tar.gz \
-    && cd pytorch-v2.0.0 \
-    && sh -c 'echo 2.0.0 > version.txt' \
-    && pip install -r requirements.txt \
-    && python3 tools/amd_build/build_amd.py \
-    && python3 setup.py install \
-    && cd ..
-RUN wget -nv https://github.com/pytorch/vision/archive/refs/tags/v0.15.1.tar.gz \
-    && tar zxf v0.15.1.tar.gz \
-    && cd vision-0.15.1 \
-    && python3 setup.py install \
-    && cd ..
-
+RUN wget -nv https://pub-1cbfe09f357e4aa1a82dc7a43cc443ab.r2.dev/wheel/torch-2.0.1+gitd0d0524-cp310-cp310-linux_x86_64.whl \
+  && pip install ./torch-2.0.1+gitd0d0524-cp310-cp310-linux_x86_64.whl
+RUN wget -nv https://pub-1cbfe09f357e4aa1a82dc7a43cc443ab.r2.dev/wheel/torchvision-0.15.2+6770a25-cp310-cp310-linux_x86_64.whl \
+  && pip install ./torchvision-0.15.2+6770a25-cp310-cp310-linux_x86_64.whl
 WORKDIR /
-# RUN rm -r /srv/src
+RUN rm -r /srv/src
 
-CMD "/bin/bash"  
+CMD "/bin/bash"