Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 363b7cd212 | |||
| 3d8d3fe84d |
@@ -0,0 +1 @@
|
|||||||
|
TheRock
|
||||||
Submodule
+1
Submodule TheRock added at b296f29cdd
@@ -67,7 +67,7 @@ RUN wget https://www.python.org/ftp/python/3.11.15/Python-3.11.15.tgz && \
|
|||||||
# Tested Working
|
# Tested Working
|
||||||
|
|
||||||
# Build ROCm latest
|
# Build ROCm latest
|
||||||
FROM python AS ROCm
|
FROM python AS rocm
|
||||||
WORKDIR /build/therock
|
WORKDIR /build/therock
|
||||||
COPY ./TheRock .
|
COPY ./TheRock .
|
||||||
ENV CFLAGS="-Wno-error=discarded-qualifiers -fcommon -w"
|
ENV CFLAGS="-Wno-error=discarded-qualifiers -fcommon -w"
|
||||||
@@ -92,7 +92,7 @@ RUN env INSTALL_PREFIX=/usr/local ./dockerfiles/install_pinned_patchelf.sh && \
|
|||||||
# Work in progress [ Breaking due to x86 intrinsics not ported to Aarch64 ]
|
# Work in progress [ Breaking due to x86 intrinsics not ported to Aarch64 ]
|
||||||
|
|
||||||
# Build pytorch
|
# Build pytorch
|
||||||
FROM ROCm AS pytorch
|
FROM rocm AS pytorch
|
||||||
# Setup Envronment
|
# Setup Envronment
|
||||||
ENV PYTORCH_ROCM_ARCH=1100
|
ENV PYTORCH_ROCM_ARCH=1100
|
||||||
ENV USE_ROCM=1
|
ENV USE_ROCM=1
|
||||||
@@ -149,5 +149,5 @@ WORKDIR /build/torchaudio
|
|||||||
RUN git clone https://github.com/pytorch/vision.git
|
RUN git clone https://github.com/pytorch/vision.git
|
||||||
|
|
||||||
# Build final stripped down image
|
# Build final stripped down image
|
||||||
FROM ubuntu:26.04 as rocm-aarch64
|
FROM ubuntu:26.04
|
||||||
# Copy in rocm, pytorch, torchaudio, torchvision
|
# Copy in rocm, pytorch, torchaudio, torchvision
|
||||||
Reference in New Issue
Block a user