This commit is contained in:
parent
687d76494c
commit
c4142cc566
|
|
@ -4,7 +4,14 @@
|
||||||
ARG CUDA_VERSION=12.6.1
|
ARG CUDA_VERSION=12.6.1
|
||||||
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 AS builder-torch
|
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 AS builder-torch
|
||||||
|
|
||||||
ENV USE_DISTRIBUTED=1
|
ENV USE_CUDA=1 \
|
||||||
|
USE_DISTRIBUTED=1 \
|
||||||
|
USE_MPI=1 \
|
||||||
|
USE_GLOO=1 \
|
||||||
|
USE_NCCL=1 \
|
||||||
|
USE_SYSTEM_NCCL=1 \
|
||||||
|
BUILD_TEST=0
|
||||||
|
|
||||||
ARG MAX_JOBS=90
|
ARG MAX_JOBS=90
|
||||||
ENV DEBIAN_FRONTEND=noninteractive \
|
ENV DEBIAN_FRONTEND=noninteractive \
|
||||||
PYTHONUNBUFFERED=1 LANG=C.UTF-8 LC_ALL=C.UTF-8 \
|
PYTHONUNBUFFERED=1 LANG=C.UTF-8 LC_ALL=C.UTF-8 \
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue