目前使用的nvidia 5090D,NVIDIA-SMI 580.95.05 Driver Version: 580.95.05 CUDA Version: 13.0
经过验证的可用的torch版本组合
torch 2.9.0+cu130 torchvision 0.24.0+cu130
python -m pip install \
torch==2.9.0+cu130 \
torchvision==0.24.0+cu130 \
--index-url https://download.pytorch.org/whl/cu130