Skip to content

Commit a775924

Browse files
authored
Update start.sh
1 parent 9198ab2 commit a775924

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

docker/base-image/start.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ function install_cuda_debian() {
4040
gpg --dearmor | tee /etc/apt/keyrings/cuda.gpg > /dev/null \
4141
&& apt-get update \
4242
&& apt-get install -y \
43-
cuda-minimal-build-${CUDA_PKG_VERSION} cuda-nvcc-${CUDA_PKG_VERSION} cuda-cudart-dev-${CUDA_PKG_VERSION} cuda-nvtx-${CUDA_PKG_VERSION}
43+
cuda-minimal-build-${CUDA_PKG_VERSION} cuda-nvcc-${CUDA_PKG_VERSION} cuda-cudart-dev-${CUDA_PKG_VERSION} cuda-nvtx-${CUDA_PKG_VERSION} cuda-nvml-${CUDA_PKG_VERSION}
4444

4545
echo "[INFO] CUDA installation complete."
4646
}

0 commit comments

Comments
 (0)