feat: set CUDA_VERSION to be 11.2.2

Meng Zhang 2023-11-05 11:42:13 -08:00
parent b4fe249636
commit 789b1f727b
1 changed files with 1 additions and 1 deletions

View File

@ -1,6 +1,6 @@
ARG UBUNTU_VERSION=22.04
# This needs to generally match the container host's environment.
ARG CUDA_VERSION=11.7.1
ARG CUDA_VERSION=11.2.2
# Target the CUDA build image
ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
# Target the CUDA runtime image