python 3.12.3 (main, Jan 22 2026, 20:57:42) [GCC 13.3.0]
torch 2.6.0+cu124 cuda 12.4 cudnn 90100 triton 3.2.0 numpy 2.2.6
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Thu_Mar_28_02:18:24_PDT_2024
Cuda compilation tools, release 12.4, V12.4.131
Build cuda_12.4.r12.4/compiler.34097967_0
1.13.2.git.kitware.jobserver-pipe-1
g++ (Ubuntu 13.3.0-6ubuntu2~24.04.1) 13.3.0
Copyright (C) 2023 Free Software Foundation, Inc.
This is free software; see the source for copying conditions.  There is NO
warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

name, driver_version, uuid, memory.total [MiB]
NVIDIA A100-SXM4-40GB, 580.126.20, GPU-900e5bea-256b-4dde-9edc-7b29a5368f63, 40960 MiB
/usr/local/cuda
/usr/local/cuda-13
/usr/local/cuda-13.0
