feat: apt clean

2024-09-10 14:19:34 -07:00
parent 2eddbb0557
commit 476aa7391c
3 changed files with 10 additions and 14 deletions
@@ -45,7 +45,7 @@ EOF
 # Install ROCm
 RUN <<EOF
 apt-get update
-apt-get install -y --no-install-recommends rocm-dev rocblas-dev
+apt-get install -y --no-install-recommends rocm-dev rocm-libs
 apt-get clean
 rm -rf /var/lib/apt/lists/*
 EOF