mmcv与cuda,pytorch版本匹配要求

这篇具有很好参考价值的文章主要介绍了mmcv与cuda,pytorch版本匹配要求。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

mmcv与cuda、pytorch版本兼容要求,见mmcv官方文档:https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html#pip 安装部分。

目前网页上默认最新版2.x版本,若要切换旧版,点击页面左下角切换即可

mmcv版本,软件环境相关,mmcv,mmlab

  1. 查看自己的cuda和torch版本:

    python -c 'import torch;print(torch.__version__);print(torch.version.cuda)'
    # pytorch 2.0版本需要cuda11.7及以上
    
  2. 点击文档链接 选择自己所需版本,拷贝对应的安装命令,进行安装
    mmcv版本,软件环境相关,mmcv,mmlab
    注意!!! open-mmlab家族已全面升级版本(1.x),与0.x有较大变化,不再兼容。mmcv中与cv无关的操作已迁至mmengine中。所以mmcv1.x与mmcv2.x版本不兼容,看一下自己的mmlab工程需要mmcv旧版还是新版
    以上链接mmcv不再可选旧版1.x版本,可根据需要手动修改安装命令中的mmcv版本。
    mmcv1.x版本号说明:
    PyTorch 在 1.x.0 和 1.x.1 之间通常是兼容的,故 mmcv 只提供 1.x.0 的编译包。如果你 的 PyTorch 版本是 1.x.1,你可以放心地安装在 1.x.0 版本编译的 mmcv。

mmcv1.x版本安装命令查询:https://mmcv.readthedocs.io/zh_CN/1.x/get_started/installation.html#pip
mmcv版本,软件环境相关,mmcv,mmlab

  1. 如果安装依赖库的时间过长,可以指定 pypi 源:

    pip install mmcv -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html -i https://pypi.tuna.tsinghua.edu.cn/simple
    
  2. opencv-python-headless说明
    如果你打算使用 opencv-python-headless 而不是 opencv-python,例如在一个很小的容器环境或者没有图形用户界面的服务器中,你可以先安装 opencv-python-headless,这样在安装 mmcv 依赖的过程中会跳过 opencv-python。文章来源地址https://www.toymoban.com/news/detail-595757.html

到了这里,关于mmcv与cuda,pytorch版本匹配要求的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • Pycharm搭建CUDA,Pytorch教程(匹配版本,安装,搭建全保姆教程)

            最近训练模型跑代码需要用到nvidia的cuda架构加速,结果网上几乎找不到什么能直接解决问题的教程,最后东拼西凑了几个小时才搭建完成,所以想整理出这篇集百家之精华的教程,防止自己以后太久不用忘记了。         首先的先知道配置好这些环境需要准备好哪

    2024年02月04日
    浏览(51)
  • Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

    本文已收录于Pytorch系列专栏: Pytorch入门与实践 专栏旨在详解Pytorch,精炼地总结重点,面向入门学习者,掌握Pytorch框架,为数据分析,机器学习及深度学习的代码能力打下坚实的基础。免费订阅,持续更新。 可能出现的问题: CUDA和cuDNN版本不匹配 CUDA和Pytorch版本不匹配

    2023年04月18日
    浏览(61)
  • 关于python pytorch 与CUDA版本相关问题

    首先在终端中输入python进入python交互式环境 官网:https://pytorch.org/get-started/previous-versions/ CUDA Toolkit版本及可用PyTorch对应关系总结(参考官网) cuda版本确定后,pytorch要依赖此cuda版本,因为pytroch安装跟cuda对应的,比如下图,torch1.11.0只适配cuda10.2、11.3;不适配cuda10.1等其他版

    2024年02月06日
    浏览(46)
  • 安装mmcv及GPU版本的pytorch及torchvision

    注意:以上适用cuda11.1版本 如果想离线安装,就看这篇文章 看这篇文章 对我的代码来说,直接装这个,注意要在anaconda powershell prompt里面装

    2024年02月07日
    浏览(44)
  • 最新版本的Anaconda环境配置、Cuda、cuDNN以及pytorch环境一键式配置流程

    本教程是最新的深度学习入门环境配置教程,跟着本教程可以帮你解决入门深度学习之前的环境配置问题。同时,本教程拒绝琐碎,大部分以图例形式进行教程。这里我们安装的都是最新版本~ 1.1 下载 首先,进入Aanconda下载地址:https://www.anaconda.com/download/ 如果嫌下载慢的话,

    2024年02月13日
    浏览(68)
  • 在Anaconda的虚拟环境上安装cuda、pytorch、opencv以及tensorflow 以及相关报错。

    首先查看自己电脑能支持的cuda版本,查看方法,命令行输入:nvidia-smi 这里我的cuda最高支持11.1的版本,下载的时候找11.1及以下的都可以 然后是在命令行进入提前创建好的虚拟环境(我的虚拟环境名字叫DLGPU,这里要换成自己的) 然后去pytorch的官网里可以找到下载cuda和对应

    2024年02月05日
    浏览(102)
  • ubuntu18.04复现yolo v8环境配置之CUDA与pytorch版本问题以及多CUDA版本安装及切换

    最近在复现yolo v8的程序,特记录一下过程 环境:ubuntu18.04+ros melodic 小知识:GPU并行计算能力高于CPU—B站UP主说的 Ubuntu可以安装多个版本的CUDA。如果某个程序的Pyorch需要不同版本的CUDA,不必删除之前的CUDA,可以实现多版本的CUDA切换 一、查看当前PyTorch使用的CUDA版本: 注意

    2024年02月11日
    浏览(52)
  • GPU版本pytorch的安装,配套环境python、Cuda、Anaconda安装和版本选择,及常见问题调用gpu返回false

    前言 :第一次装这个我也很懵,就想记录一下交流经验,这个安装最麻烦的是需要各个 版本 都需要 对应 。我也看了很多教程网上基本上安装都是cpu版本,就官网链接安装下来也是cpu版本,然后就不能调用显卡。 本教程使用python3.9、pytorch1.8.0、Cuda11.2版本、Cudnn8.8.1,这个

    2024年02月03日
    浏览(53)
  • 学习笔记:在Anaconda环境下装Pytorch,CUDA,pillow,numpy等库并寻找适配的Python版本

    目录 1.在Anaconda Prompt创建新环境 2.去pytorch官网上查找环境中Python所对应的pytorch版本并下载相关包 查询电脑独立NVIDIA显卡所适配的CUDA版本 CUDA版本查询 3.部分库版本的安装与修改  本人在Anaconda下创建的新环境为 python 3.7.0 pytorch 1.8.0 pillow 9.5.0 numpy 1.21.5 能够正常运行 如果我这

    2024年02月05日
    浏览(45)
  • 集显安装pytorch教程,cuda和cudann环境配置torch-GPU版本看这一篇就够了

    集显看最下面 首先GPU安装教程 1.安装Anaconda (这个是可以安装环境和前置的软件),这个网址是清华源下载,可以去官网下载 Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror 2.安装CUDA (这个是显卡驱动,需要安装好以用显卡计算,缩短运行时间)(c

    2024年02月02日
    浏览(75)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包