1
0
forked from eoeair/cenv
This commit is contained in:
ben0i0d 2025-02-11 14:20:33 +08:00
parent ae0b9b0596
commit a807edf478

View File

@ -5,9 +5,9 @@ ARG VERSION=6.3.1
ENV PATH="${PATH}:/opt/rocm/bin"
RUN apt-get update && apt-get install -y --no-install-recommends ca-certificates wget gnupg2 && \
wget https://repo.radeon.com/amdgpu-install/6.3.1/ubuntu/jammy/amdgpu-install_6.3.60302-1_all.deb && \
wget https://repo.radeon.com/amdgpu-install/6.3.1/ubuntu/jammy/amdgpu-install_6.3.60301-1_all.deb && \
# add ROCm
apt-get install -y ./amdgpu-install_6.3.60302-1_all.deb && rm ./amdgpu-install_6.3.60302-1_all.deb && \
apt-get install -y ./amdgpu-install_6.3.60301-1_all.deb && rm ./amdgpu-install_6.3.60301-1_all.deb && \
# Install ROCm-ml-libraries
apt-get update && apt-get update && apt-get install -y rocm-ml-libraries && \
apt-get clean && rm -rf /var/lib/apt/lists/*