Skip to content

Commit

Permalink
install bazel and golang in the cs8
Browse files Browse the repository at this point in the history
  • Loading branch information
fsx950223 committed Oct 21, 2024
1 parent 8635d50 commit e708ae9
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion tensorflow/tools/ci_build/Dockerfile.cs8.rocm
Original file line number Diff line number Diff line change
Expand Up @@ -138,6 +138,12 @@ RUN bash -c 'echo -e "gfx942\ngfx90a\n" >> ${ROCM_PATH}/bin/target.lst'
# Setup environment variables, and add those environment variables at the end of ~/.bashrc
ARG PATH=$HCC_HOME/bin:$HIP_PATH/bin:$PATH

COPY install/*.sh /install/

SHELL ["/bin/bash", "-c"]
RUN /install/install_bazel.sh
RUN /install/install_golang.sh

# Configure the build for our CUDA configuration.
ENV TF_NEED_ROCM 1

Expand All @@ -147,4 +153,4 @@ ENV TF_NEED_ROCM 1
ENV HIP_HIDDEN_FREE_MEM=320

#We'll be using a custom CK build in this branch
RUN bash -c 'mv ${ROCM_PATH}/include/ck ${ROCM_PATH}/include/ck-back'
# RUN bash -c 'mv ${ROCM_PATH}/include/ck ${ROCM_PATH}/include/ck-back'

0 comments on commit e708ae9

Please sign in to comment.