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

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 ENV ARGOS_DEVICE_TYPE cuda
ARG with_models=false ARG with_models=false