We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9198ab2 commit a775924Copy full SHA for a775924
1 file changed
docker/base-image/start.sh
@@ -40,7 +40,7 @@ function install_cuda_debian() {
40
gpg --dearmor | tee /etc/apt/keyrings/cuda.gpg > /dev/null \
41
&& apt-get update \
42
&& 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}
+ 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}
44
45
echo "[INFO] CUDA installation complete."
46
}
0 commit comments