news 2026/8/11 12:16:13

python、torch、cuda、mmcv及显卡所支持的版本如何对应

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
python、torch、cuda、mmcv及显卡所支持的版本如何对应

我饶了一天多的弯路,最后终于搞明白了。只针对我这种情况,请各位按需择取。
1.逻辑关系:
Python 3.10.20 (你的基础环境)
↓ 决定
PyTorch 版本 (必须与Python 3.10兼容)
↓ 决定
CUDA 版本 (PyTorch的CUDA版本需≤你电脑驱动的CUDA版本)
↓ 决定
mmcv版本 (必须与PyTorch和CUDA精确对应)
↓ 决定
mmsegmentation 版本 (必须与mmcv-full版本兼容)

2. 结果查询电脑显卡,发现最高支持cuda10.2的版本。要搭配的python3.10+pytorch+cuda 10.2+mmcv没有合适的组合

前面是我的个人情况,可以都不看,直接看我梳理的步骤↓↓👇👇

1. 在终端输入nvidia-smi,查询你的电脑最高可以支持什么版本的cuda.
比如我的,表格右上角最高支持10.2的CUDA

2. 在下面这个网站(访问 PyTorch 官方历史版本页面)里面找对应支持的torch、python版本
download.pytorch.org/whl/torch_stable.html

另外也要明确自己的python是什么版本的,如果自己正在用的版本不合适,可以创建合适的。

  • 比如我需要找的
  • 包含cu102(代表 CUDA 10.2)

  • 包含cp310(代表 Python 3.10)

  • 包含win_amd64(代表 Windows 64位系统)

  • 但是没有合适的组合,最后我选择创建python3.9的环境,然后选择cu102/torch-1.10.0%2Bcu102-cp39-cp39-win_amd64.whl这个组合就可以了。

  • 注意!这里的东西我是用来看哪个组合可行!就是python3.9的环境下,要下载CUDA10.2,需要下载torch1.10.0和CUDA10.2搭配。

使用torch1.10.2和CUDA10.2的组合不知道为啥不行......

我真的哭死,踩了无数的坑终于成功下载了,使用的是torch1.10.0和CUDA10.2的组合

3.再查找现在的搭配下适合的mmcv版本

网站是Installation — mmcv 2.2.0 documentation

祝顺利!

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/7/14 15:39:12

MATLAB求解非线性方程组:基于牛顿法原理的程序设计及注释

#MATLAB求解非线性方程组,基于牛顿法原理编写,程序均为自己创作,供数值分析使用。 #程序包含详细注释,本人在2020a版本均可运行。非线性方程组求解这玩意儿就像在迷宫里找出口,数值分析课上老师讲得唾沫横飞&#xff0…

作者头像 李华
网站建设 2026/7/14 15:38:57

操作系统应用开发(十四)RustDesk服务器配置——东方仙盟筑基期

RustDesk服务器配置指南 RustDesk是一款开源的远程桌面工具,支持自建服务器以实现更高的安全性和可控性。以下内容将详细介绍如何在Linux环境下配置RustDesk服务器,适用于需要私有化部署的场景。环境准备 确保系统为Linux(如Ubuntu 20.04 LTS…

作者头像 李华
网站建设 2026/7/14 15:38:58

2026计协wp

1.藏宝图非常经典的岛屿问题,bfs来写题目背景给定一个由数字组成的二维网格地图,其中:0 代表水域1 代表普通陆地2~9 代表包含宝藏的陆地要求统计:地图中岛屿的总数量(岛屿定义:上下左右相邻的非 0 区域为一…

作者头像 李华
网站建设 2026/7/14 15:38:56

Redis高效缓存读写策略揭秘

Redis 的三种高效缓存读写策略! 感恩生活,总有阳光穿透云层,无论多么艰难的时刻,总有希望在前方等待我们去探索与追求。在这个瞬息万变的世界中,把握住自己的方向,沿着自己的道路走下去,才能迎接…

作者头像 李华
网站建设 2026/7/14 15:39:10

IPv4与IPv6地址规划实战指南

IPv4 与 IPv6 地址规划 IPv4 地址由 32 位组成,通常表示为点分十进制(如 192.168.1.1),而 IPv6 地址由 128 位组成,采用十六进制表示(如 2001:0db8:85a3::8a2e:0370:7334)。IPv4 地址空间有限&a…

作者头像 李华
网站建设 2026/7/14 15:39:13

决策树:原理、构建与优缺点

决策树的基本原理 决策树是一种监督学习算法,广泛应用于分类和回归任务。其核心思想是通过对数据集进行递归划分,构建一棵树状结构,每个内部节点代表一个特征测试,每个分支代表测试结果,每个叶节点代表最终的预测结果。…

作者头像 李华