chore: formatting

2024-09-10 13:56:19 -07:00
parent 49238807cd
commit 9dc3b7ac97
9 changed files with 8 additions and 191 deletions
@@ -1,4 +1,7 @@
-FROM wtongze/rocm-base:5.7.3
+ARG ROCM_VERSION="5.7.3"
+
+FROM wtongze/rocm-base:${ROCM_VERSION}
+
 ARG TENSORFLOW_BRANCH="r2.11-rocm-enhanced"
 ARG PYTHON_VERSION="3.11"
 ARG NUMPY_VERSION="1.22.4"
@@ -32,7 +35,7 @@ EOF
 ENV PATH=$PATH:/home/rocm/.local/bin

 # TODO: simplify dependencies
-# > https://github.com/ROCm/ROCm/issues/1880
+# > Ref: https://github.com/ROCm/ROCm/issues/1880
 RUN python${PYTHON_VERSION} -m pip install -U --user pip numpy==${NUMPY_VERSION} wheel packaging requests opt_einsum patchelf
 RUN python${PYTHON_VERSION} -m pip install -U --user keras_preprocessing --no-deps