Update readme

This commit is contained in:
Ying Sheng
2024-07-25 08:14:36 -07:00
parent 1a491d00cb
commit 7802df1e2b
2 changed files with 4 additions and 5 deletions

View File

@@ -1,8 +1,7 @@
ARG CUDA_VERSION=12.4.1
ARG CUDA_VERSION=12.1.1
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
ARG CUDA_VERSION=12.4.1
ARG PYTHON_VERSION=3
ENV DEBIAN_FRONTEND=noninteractive