RTX50 系列显卡与 CUDA、PyTorch 及 Python 版本对应关系

背景说明

本文基于 RTX5070 显卡测试，记录其对 CUDA、PyTorch 和 Python 版本的兼容性要求。

CUDA 版本要求

RTX50 系列显卡目前仅支持 CUDA 12.8。

PyTorch 版本要求

目前仅支持 PyTorch 2.7.0 和 2.7.1。

可访问 PyTorch 官网查看对应 CUDA 12.8 的 wheel 文件，根据操作系统和 Python 版本选择。

torch 2.7.1 及对应 torchvision、torchaudio 安装指令：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

Python 版本要求

Python 版本不能过低，建议至少使用 3.9 及以上版本。

注意事项

PyTorch 通常向下兼容，高版本可用于运行低版本项目，但可能会遇到 numpy 等库的兼容性问题，需适当调整。