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

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等库有点问题得改改。

Read more

【基础算法】算法的“预谋”:前缀和如何改变游戏规则

【基础算法】算法的“预谋”:前缀和如何改变游戏规则

🔭 个人主页:散峰而望 《C语言:从基础到进阶》《编程工具的下载和使用》《C语言刷题》《算法竞赛从入门到获奖》《人工智能》《AI Agent》 愿为出海月,不做归山云 🎬博主简介 【基础算法】算法的“预谋”:前缀和如何改变游戏规则 * 前言 * 前缀和 * 1.1 一维前缀和 * 1.1.1 前缀和 * 1.1.2 最大子段和 * 1.2 二维前缀和 * 1.2.1 二维前缀和 * 1.2.2 激光炸弹 * 结语 前言 在算法设计与优化中,前缀和是一种简单却强大的技巧,能够将复杂问题转化为高效计算。无论是处理一维数组的区间求和,还是解决二维矩阵的子矩阵问题,前缀和都能通过预处理将时间复杂度从线性降低到常数级别,彻底改变问题的解决方式。

【强化学习】Double DQN(Double Deep Q-Network)算法

【强化学习】Double DQN(Double Deep Q-Network)算法

📢本篇文章是博主强化学习(RL)领域学习时,用于个人学习、研究或者欣赏使用,并基于博主对相关等领域的一些理解而记录的学习摘录和笔记,若有不当和侵权之处,指出后将会立即改正,还望谅解。文章分类在👉强化学习专栏:        【强化学习】- 【单智能体强化学习】(12)---《Double DQN(Double Deep Q-Network)算法》 Double DQN(Double Deep Q-Network)算法 目录 一、Double DQN算法详解 二、算法背景和提出 2.1 过估计偏差问题 2.2 Double Q-Learning的灵感 2.3 Double DQN的提出 三、Double DQN的核心思想 四、算法流程

数据结构—顺序表链表超经典算法

数据结构—顺序表链表超经典算法

数据结构—顺序表链表经常用到的算法 * 所有题目链接 * 顺序表算法题(双指针法) * 移除元素 * 删除有序数组中的重复项 * 合并两个有序数组 * 链表算法题(快慢指针,三指针法,创建新链表法) * 移除链表元素 * 反转链表 * 链表的中间节点 * 合并两个有序链表 * 链表分割 * 链表的回文结构 * 相交链表 * 环形链表(快慢指针) * 环形链表I * 环形链表II * 代码仓库 所有题目链接 移除元素 删除有序数组中的重复项 合并两个有序数组 移除链表元素 反转链表 链表的中间节点 合并两个有序链表 链表分割 链表的回文结构 相交链表 环形链表I 环形链表II 顺序表算法题(双指针法) 移除元素 题目链接↓ 移除元素 题目讲解↓ 思路:双指针法,创建两个变量dst,src如果src指向的数据是val,src++如果src指向的数据不是val,赋值(

【优选算法 | 二分查找】二分查找算法解析:如何通过二段性优化搜索效率

【优选算法 | 二分查找】二分查找算法解析:如何通过二段性优化搜索效率

算法相关知识点可以通过点击以下链接进行学习一起加油!双指针滑动窗口 在本篇文章中,我们将深入解析二分查找算法的核心原理。从基本概念到实际应用,带你了解如何利用二分查找高效定位元素,提升搜索效率。无论你是刚接触算法的新手,还是想优化代码性能的老手,二分查找都是你不可忽视的强大工具! 🌈个人主页:是店小二呀 🌈C/C++专栏:C语言\ C++ 🌈初/高阶数据结构专栏: 初阶数据结构\ 高阶数据结构 🌈Linux专栏: Linux 🌈算法专栏:算法 🌈Mysql专栏:Mysql 🌈你可知:无人扶我青云志 我自踏雪至山巅 文章目录 * 34. 在排序数组中查找元素的第一个和最后一个位置(重要) * 二段性(重要/必备知识) * 1.查找左端点 * 2.循环判断条件 * 3. left和right移动方式 * 3.求中点操作 * 4.总结二分模板 * 69.x 的平方根