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

fix: tf gfx check

Browse Source
This commit is contained in:
wtongze
2024-09-11 19:58:43 -07:00
parent 7bd4c10b51
commit bd42ce0763
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tensorflow.dockerfile
+2
View File
@@ -47,6 +47,8 @@ RUN python${PYTHON_VERSION} -m pip install -U --user keras_preprocessing --no-de
RUN sed -i -e "s/python3/python${PYTHON_VERSION}/g" -e "s/pip3/python${PYTHON_VERSION} -m pip/g" build_rocm_python3 RUN sed -i -e "s/python3/python${PYTHON_VERSION}/g" -e "s/pip3/python${PYTHON_VERSION} -m pip/g" build_rocm_python3
RUN sudo ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python RUN sudo ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python
RUN sed -i 's/"gfx900",/"gfx803", "gfx900",/g' tensorflow/compiler/xla/stream_executor/device_description.h
ENV ROCM_PATH=/opt/rocm-${ROCM_VERSION} ENV ROCM_PATH=/opt/rocm-${ROCM_VERSION}
ENV ROCM_TOOLKIT_PATH=${ROCM_PATH} ENV ROCM_TOOLKIT_PATH=${ROCM_PATH}
RUN ./build_rocm_python3 ${ROCM_PATH} RUN ./build_rocm_python3 ${ROCM_PATH}