fix dockerfile and base builder (#1795) [skip-ci]

This commit is contained in:
Wing Lian
2024-07-30 08:34:37 -04:00
committed by GitHub
parent d8d1788ffc
commit d4f6a6b103
2 changed files with 5 additions and 1 deletions

View File

@@ -3,7 +3,7 @@ ARG CUDNN_VERSION="8"
ARG UBUNTU_VERSION="22.04"
ARG MAX_JOBS=4
FROM nvidia/cuda:$CUDA_VERSION-cudnn$CUDNN_VERSION-devel-ubuntu$UBUNTU_VERSION as base-builder
FROM nvidia/cuda:$CUDA_VERSION-cudnn$CUDNN_VERSION-devel-ubuntu$UBUNTU_VERSION AS base-builder
ENV PATH="/root/miniconda3/bin:${PATH}"