Use the torch build whls from evshiron/rocm_lab instead of building it

from sources
chirvo · May 12, 2023 · 64e492f · 64e492f
1 parent 5861b29
commit 64e492f
Showing 1 changed file with 7 additions and 17 deletions.
diff --git 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"