Added support for specifying targets for docker build

TabbyML · Jan 14, 2024 · b13bbec · b13bbec
1 parent b471554
commit b13bbec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rocm.Dockerfile b/rocm.Dockerfile
@@ -6,6 +6,8 @@ ARG AMDGPU_INSTALL_URL=https://repo.radeon.com/amdgpu-install/6.0/ubuntu/${UBUNT
 FROM ubuntu:${UBUNTU_VERSION} as hipblas_base
 
 ARG AMDGPU_INSTALL_URL
+ARG AMDGPU_TARGETS
+ENV PATH="/opt/rocm/bin:${PATH}"
 
 # Install ROCm
 RUN apt-get update &&  \