fix default base image
This commit is contained in:
@@ -6,7 +6,7 @@ ARG MAX_JOBS=4
|
|||||||
ARG BASE_IMAGE=nvidia/cuda
|
ARG BASE_IMAGE=nvidia/cuda
|
||||||
ARG DEFAULT_TAG=${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu${UBUNTU_VERSION}
|
ARG DEFAULT_TAG=${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu${UBUNTU_VERSION}
|
||||||
ARG BASE_TAG=""
|
ARG BASE_TAG=""
|
||||||
FROM ${BASE_IMAGE}:${BASE_TAG:-${DEFAULT_TAG}} AS base-builder
|
FROM ${BASE_IMAGE:-nvidia/cuda}:${BASE_TAG:-${DEFAULT_TAG}} AS base-builder
|
||||||
|
|
||||||
ENV PATH="/root/miniconda3/bin:${PATH}"
|
ENV PATH="/root/miniconda3/bin:${PATH}"
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user