ml-playground-torch

测试环境

Thu Nov 20 19:38:50 2025
+-----------------------------------------------------------------------------+ | NVIDIA-SMI 450.191.01 Driver Version: 450.191.01 CUDA Version: 11.0 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | | | | MIG M. | |===============================+======================+======================| | 0 Tesla T4 On | 00000000:3D:00.0 Off | 0 | | N/A 59C P0 36W / 70W | 0MiB / 15109MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+ | 1 Tesla T4 On | 00000000:3E:00.0 Off | 0 | | N/A 67C P0 40W / 70W | 0MiB / 15109MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+ | 2 Tesla T4 On | 00000000:40:00.0 Off | 0 | | N/A 58C P0 36W / 70W | 0MiB / 15109MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+ | 3 Tesla T4 On | 00000000:41:00.0 Off | 0 | | N/A 58C P0 35W / 70W | 0MiB / 15109MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+ | 4 Tesla T4 On | 00000000:B1:00.0 Off | 0 | | N/A 56C P0 35W / 70W | 0MiB / 15109MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+ | 5 Tesla T4 On | 00000000:B2:00.0 Off | 0 | | N/A 58C P0 36W / 70W | 0MiB / 15109MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+ | 6 Tesla T4 On | 00000000:B4:00.0 Off | 0 | | N/A 59C P0 36W / 70W | 0MiB / 15109MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+ | 7 Tesla T4 On | 00000000:B5:00.0 Off | 0 | | N/A 59C P0 35W / 70W | 0MiB / 15109MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+ | Processes: | | GPU GI CI PID Type Process name GPU Memory | | ID ID Usage | |=============================================================================| | No running processes found | +-----------------------------------------------------------------------------+

CUDA 和 Torch 依赖安装

mamba install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia