feat: simplify rocm 4.5+

This commit is contained in:
2024-09-11 21:01:57 -07:00
parent bd42ce0763
commit 2b2d578ab8
3 changed files with 8 additions and 16 deletions
-8
View File
@@ -29,14 +29,6 @@ EOF
RUN python${PYTHON_VERSION} -m pip install typing-extensions wheel
# For ROCm 4.5+
RUN <<EOF
sudo apt-get update
sudo apt-get install -y rocm-ml-sdk rocm-hip-sdk
sudo apt-get clean
sudo rm -rf /var/lib/apt/lists/*
EOF
RUN <<EOF
sudo apt-get update
sudo apt-get install -y ninja-build