fix: use devel for Triton's compiler requirements (#1074)

This commit is contained in:
Yineng Zhang
2024-08-13 19:08:43 +08:00
committed by GitHub
parent f7fb68d292
commit 0076f11541

View File

@@ -1,5 +1,5 @@
ARG CUDA_VERSION=12.1.1
FROM nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu20.04
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
ARG BUILD_TYPE=all
ENV DEBIAN_FRONTEND=noninteractive