查看版本：
# nvidia-smi Wed Apr 15 17:56:12 2026 +-----------------------------------------------------------------------------------------+ | NVIDIA-SMI 570.211.01 Driver Version: 570.211.01 CUDA Version: 12.8 | |-----------------------------------------+------------------------+----------------------+ | 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 NVIDIA GeForce RTX 3080 Ti Off | 00000000:00:03.0 Off | N/A | | 30% 26C P8 6W / 350W | 1MiB / 12288MiB | 0% Default | | | | N/A | +-----------------------------------------+------------------------+----------------------+ +-----------------------------------------------------------------------------------------+ | Processes: | | GPU GI CI PID Type Process name GPU Memory | | ID ID Usage | |=========================================================================================| | No running processes found | +-----------------------------------------------------------------------------------------+
安装对应版本：
根据上面的信息将 cu118 换成 cu128
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
地址为：
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128