Skip to content

Commit 54a97b7

Browse files
committed
Accelerated Python: Install newer versions of ncu and nsys.
1 parent abcf126 commit 54a97b7

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tutorials/accelerated-python/brev/dockerfile

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,15 @@ RUN apt-get update -y \
2929
&& apt-get clean -y \
3030
&& rm -rf /var/lib/apt/lists/*
3131

32+
# Install Nsight Systems and Nsight Compute.
33+
RUN curl -L -O https://developer.nvidia.com/downloads/assets/tools/secure/nsight-systems/2025_3/NsightSystems-linux-cli-public-2025.3.1.90-3582212.deb \
34+
&& dpkg -i NsightSystems-linux-cli-public-2025.3.1.90-3582212.deb \
35+
&& rm -f NsightSystems-linux-cli-public-2025.3.1.90-3582212.deb \
36+
&& curl -L -O https://developer.download.nvidia.com/compute/cuda/repos/debian12/x86_64/nsight-compute-2025.2.1_2025.2.1.3-1_amd64.deb \
37+
&& dpkg -i nsight-compute-2025.2.1_2025.2.1.3-1_amd64.deb \
38+
&& rm -f nsight-compute-2025.2.1_2025.2.1.3-1_amd64.deb \
39+
&& update-alternatives --install /usr/bin/ncu ncu /opt/nvidia/nsight-compute/2025.2.1/ncu 20250201
40+
3241
# Install PyTorch with CUDA 12.8 support.
3342
RUN pip install --no-cache-dir --root-user-action=ignore \
3443
--index-url https://download.pytorch.org/whl/cu128 \

0 commit comments

Comments
 (0)