Small build fixes.

nvidia-riva · Sep 7, 2023 · 55b6e4d · 55b6e4d
1 parent a3b5bbd
commit 55b6e4d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -3,10 +3,10 @@
 ## It is possible to just use defaults and have a pure manylinux2014 with CUDA on top
 ## DALI is based on "manylinux2014", official page https://github.com/pypa/manylinux
 #########################################################################################
-ARG CUDA_VER=11.2.1
+ARG CUDA_VER=11.2.2
 ARG ARCH=x86_64
 ARG FROM_IMAGE_NAME=quay.io/pypa/manylinux2014_x86_64
-ARG CUDA_IMAGE=nvidia/cuda:11.2.1-devel-centos7 # nvidia/dali:cuda${CUDA_VER}_${ARCH}.toolkit
+ARG CUDA_IMAGE=nvidia/cuda:${CUDA_VER}-devel-centos7 # nvidia/dali:cuda${CUDA_VER}_${ARCH}.toolkit
 FROM ${CUDA_IMAGE} as cuda
 FROM ${FROM_IMAGE_NAME}
 

diff --git a/cibuildwheel.sh b/cibuildwheel.sh
@@ -8,4 +8,6 @@ export CIBW_BUILD_VERBOSITY=3
 
 docker build -t $CIBW_MANYLINUX_X86_64_IMAGE .
 
+rm -rf build/
+
 pipx run cibuildwheel --platform linux