Skip to content

Commit 17a9aa5

Browse files
authored
CUDA NVTX installation added: #21
1 parent fae1d7b commit 17a9aa5

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
@@ -39,7 +39,7 @@ function install_cuda_debian() {
3939
gpg --dearmor | tee /etc/apt/keyrings/cuda.gpg > /dev/null \
4040
&& apt-get update \
4141
&& apt-get install -y \
42-
cuda-minimal-build-${CUDA_PKG_VERSION} cuda-nvcc-${CUDA_PKG_VERSION} cuda-cudart-dev-${CUDA_PKG_VERSION}
42+
cuda-minimal-build-${CUDA_PKG_VERSION} cuda-nvcc-${CUDA_PKG_VERSION} cuda-cudart-dev-${CUDA_PKG_VERSION} cuda-nvtx-12-${CUDA_PKG_VERSION}
4343

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

0 commit comments

Comments
 (0)