Shawn732 commited on
Commit
6445214
·
1 Parent(s): 123841a

7th7th7th7th7th7th7th

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
  # Use an NVIDIA CUDA base image
2
- ARG CUDA_IMAGE="12.1.1-devel-ubuntu22.04"
3
  FROM nvidia/cuda:${CUDA_IMAGE}
4
 
5
  ENV HOST 0.0.0.0
 
1
  # Use an NVIDIA CUDA base image
2
+ ARG CUDA_IMAGE="12.3.1-base-ubuntu22.04"
3
  FROM nvidia/cuda:${CUDA_IMAGE}
4
 
5
  ENV HOST 0.0.0.0