RTX50系显卡+CUDA+torch+python对应关系

前言

本人的显卡是RTX5070，使用时发现它对CUDA、torch和python版本有要求，试图按照老项目的依赖文件进行安装发现安不了，因此记录一下（截至2025年6月）。

提示：以下是本篇文章正文内容，下面案例可供参考

一、RTX50系显卡只能使用CUDA12.8

二、目前只支持torch2.7.0和2.7.1

1.去pytorch官网的https://download.pytorch.org/whl/torch/看一下，确实只有2.7.0和2.7.1有对应cuda12.8的。

注：此图中包含了所有对应CUDA12.8的whl文件，可根据操作系统和python版本选择

torch2.7.1及对应torchvision torchaudio安装指令：

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

三、同理上图，python版本太低也不行，至少得3.9以上吧

总结

不过torch这些一般是向下兼容的，一般可以直接用高版本的去跑老的低torch版本的项目，但可能会遇到numpy等库有点问题得改改。