File tree Expand file tree Collapse file tree 1 file changed +9
-0
lines changed
tutorials/accelerated-python/brev Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Original file line number Diff line number Diff 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.
3342RUN pip install --no-cache-dir --root-user-action=ignore \
3443 --index-url https://download.pytorch.org/whl/cu128 \
You can’t perform that action at this time.
0 commit comments