feat: simplify rocm 4.5+
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user