diff --git a/env_setup/Dockerfile b/env_setup/Dockerfile index 55cb59036db3c3043e866d6c90349cb874004521..1aff95e85ca1912ea5f0778b14837bd9e2514d32 100644 --- a/env_setup/Dockerfile +++ b/env_setup/Dockerfile @@ -1,7 +1,7 @@ ############################################################################### # Nvidia already has a docker image where they installed all their drivers ############################################################################### -FROM nvidia/cuda:11.8.0-devel-ubuntu22.04 +FROM nvidia/cuda:12.2.2-devel-ubuntu22.04 ENV CUDA_VISIBLE_DEVICES=0 ###############################################################################