feat: set CUDA_VERSION to be 11.2.2
parent
b4fe249636
commit
789b1f727b
|
|
@ -1,6 +1,6 @@
|
||||||
ARG UBUNTU_VERSION=22.04
|
ARG UBUNTU_VERSION=22.04
|
||||||
# This needs to generally match the container host's environment.
|
# 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
|
# Target the CUDA build image
|
||||||
ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
|
ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
|
||||||
# Target the CUDA runtime image
|
# Target the CUDA runtime image
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue