We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fae1d7b commit 17a9aa5Copy full SHA for 17a9aa5
1 file changed
docker/base-image/start.sh
@@ -39,7 +39,7 @@ function install_cuda_debian() {
39
gpg --dearmor | tee /etc/apt/keyrings/cuda.gpg > /dev/null \
40
&& apt-get update \
41
&& apt-get install -y \
42
- cuda-minimal-build-${CUDA_PKG_VERSION} cuda-nvcc-${CUDA_PKG_VERSION} cuda-cudart-dev-${CUDA_PKG_VERSION}
+ cuda-minimal-build-${CUDA_PKG_VERSION} cuda-nvcc-${CUDA_PKG_VERSION} cuda-cudart-dev-${CUDA_PKG_VERSION} cuda-nvtx-12-${CUDA_PKG_VERSION}
43
44
echo "[INFO] CUDA installation complete."
45
}
0 commit comments