[CI] Fix Docker build for CUDA 11 (#6202)
This commit is contained in:
committed by
GitHub
parent
fd58005edf
commit
f121f2738f
@@ -1,5 +1,6 @@
|
||||
ARG CUDA_VERSION
|
||||
FROM nvidia/cuda:$CUDA_VERSION-devel-ubuntu18.04
|
||||
ARG CUDA_VERSION_ARG
|
||||
FROM nvidia/cuda:$CUDA_VERSION_ARG-devel-ubuntu18.04
|
||||
ARG CUDA_VERSION_ARG
|
||||
|
||||
# Environment
|
||||
ENV DEBIAN_FRONTEND noninteractive
|
||||
|
||||
Reference in New Issue
Block a user