wtongze/rocm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: change arch to gfx803 only

Browse Source
This commit is contained in:
wtongze
2024-09-13 09:32:19 -07:00
parent af09aa8e34
commit 0abbf5d2e5
4 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
base.dockerfile
+2 -2
View File
@@ -5,8 +5,8 @@ FROM ubuntu:22.04 AS base
ARG ROCM_VERSION="5.7.3" ARG ROCM_VERSION="5.7.3"
ARG ROCM_APT_CODE_NAME="jammy" ARG ROCM_APT_CODE_NAME="jammy"
ARG ROCBLAS_BRANCH="release/rocm-rel-5.7" ARG ROCBLAS_BRANCH="release/rocm-rel-5.7"
# gfx803 as single target may cause error # "gfx803" or "gfx803;gfx900"
ARG ROCBLAS_ARCH="gfx803;gfx900" ARG ROCBLAS_ARCH="gfx803"
ARG DEBIAN_FRONTEND="noninteractive" ARG DEBIAN_FRONTEND="noninteractive"
# Install essentials # Install essentials
blender.dockerfile
+2 -1
View File
@@ -2,7 +2,8 @@ ARG ROCM_VERSION="5.7.3"
FROM wtongze/rocm-base:${ROCM_VERSION} FROM wtongze/rocm-base:${ROCM_VERSION}
ARG HIP_ARCH="gfx803;gfx900" # "gfx803" or "gfx803;gfx900"
ARG HIP_ARCH="gfx803"
ARG BLENDER_BRANCH="v3.6.15" ARG BLENDER_BRANCH="v3.6.15"
ARG DEBIAN_FRONTEND="noninteractive" ARG DEBIAN_FRONTEND="noninteractive"
pytorch.dockerfile
+2 -1
View File
@@ -7,7 +7,8 @@ ARG PYTORCH_BRANCH="v2.2.2"
ARG TORCHVISION_BRANCH="v0.17.2" ARG TORCHVISION_BRANCH="v0.17.2"
ARG PYTHON_VERSION="3.11" ARG PYTHON_VERSION="3.11"
ARG NUMPY_VERSION="1.26.4" ARG NUMPY_VERSION="1.26.4"
ARG HIP_ARCH="gfx803;gfx900" # "gfx803" or "gfx803;gfx900"
ARG HIP_ARCH="gfx803"
ARG DEBIAN_FRONTEND="noninteractive" ARG DEBIAN_FRONTEND="noninteractive"
# Add GPU target # Add GPU target
tensorflow.dockerfile
+3 -2
View File
@@ -6,7 +6,8 @@ ARG ROCM_VERSION="5.7.3"
ARG TENSORFLOW_BRANCH="r2.12-rocm-enhanced" ARG TENSORFLOW_BRANCH="r2.12-rocm-enhanced"
ARG PYTHON_VERSION="3.11" ARG PYTHON_VERSION="3.11"
ARG NUMPY_VERSION="1.24.3" ARG NUMPY_VERSION="1.24.3"
ARG HIP_ARCH="gfx803;gfx900" # "gfx803" or "gfx803;gfx900"
ARG HIP_ARCH="gfx803"
ARG DEBIAN_FRONTEND="noninteractive" ARG DEBIAN_FRONTEND="noninteractive"
# Add GPU target # Add GPU target
@@ -46,7 +47,7 @@ ENV ROCM_TOOLKIT_PATH=${ROCM_PATH}
RUN ./build_rocm_python3 ${ROCM_PATH} RUN ./build_rocm_python3 ${ROCM_PATH}
WORKDIR /tmp/tensorflow_pkg WORKDIR /tmp/tensorflow_pkg
RUN cp $(ls -l tensorflow*.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/tensorflow-rocm.whl RUN cp $(ls -l tensorflow*.whl | awk '{ print $9 }' | tail -n1 | tr -d '\n') /home/rocm/
WORKDIR /home/rocm WORKDIR /home/rocm
CMD [ "/bin/bash", "-l" ] CMD [ "/bin/bash", "-l" ]