Update cuda.Dockerfile to CUDA 12

This commit is contained in:
Savion Lee 2024-05-06 14:06:40 -05:00 committed by GitHub
parent 67c5348c02
commit 16ae59aa43
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -1,4 +1,4 @@
FROM nvidia/cuda:11.2.2-devel-ubuntu20.04
FROM nvidia/cuda:12.4.1-devel-ubuntu20.04
ENV ARGOS_DEVICE_TYPE cuda
ARG with_models=false