yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-07-01 01:37:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

feat: update Dockerfile for arm64 build

Browse Source
This commit is contained in:
Junya Morioka
2025-12-21 11:03:33 +09:00
parent b532b8593f
commit 232329fd63
2 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
self-hosted-runner/Dockerfile
+3 -4
View File
@@ -52,9 +52,6 @@ RUN usermod -aG sudo ubuntu \
# Allow sudo without password
RUN echo '%sudo ALL=(ALL) NOPASSWD: ALL' >> /etc/sudoers
COPY entrypoint.sh /entrypoint.sh
RUN chmod +x /entrypoint.sh
WORKDIR /home/ubuntu
USER ubuntu
@@ -66,6 +63,8 @@ USER root
RUN ./bin/installdependencies.sh
USER ubuntu
COPY entrypoint.sh /entrypoint.sh
RUN chmod +x /entrypoint.sh
ENTRYPOINT ["/entrypoint.sh"]