diff --git a/ci-wheel.Dockerfile b/ci-wheel.Dockerfile index ec3ec8c..1656fc8 100644 --- a/ci-wheel.Dockerfile +++ b/ci-wheel.Dockerfile @@ -1,6 +1,5 @@ ARG CUDA_VER=11.8.0 ARG LINUX_VER=ubuntu20.04 -ARG REAL_ARCH=x86_64 ARG BASE_IMAGE=nvcr.io/nvidia/cuda:${CUDA_VER}-devel-${LINUX_VER} FROM ${BASE_IMAGE} @@ -8,7 +7,7 @@ FROM ${BASE_IMAGE} ARG CUDA_VER ARG LINUX_VER ARG CPU_ARCH -ARG REAL_ARCH +ARG REAL_ARCH=x86_64 ARG PYTHON_VER=3.9 ARG MANYLINUX_VER ARG POLICY=${MANYLINUX_VER}