Linux下編譯MMCV

介紹

MMCV 有兩個主要版本,mmcv 和 mmcv-lite。

mmcv 是全面版本,包含所有的特性以及豐富的開箱即用的 CUDA 算子,但構建時間較長;mmcvlite 是精簡版,不包含 CUDA 算子,但擁有其他所有功能,適用于不需要 CUDA 算子的場景,類似于 MMCV 早期版本(mmcv < 1.0.0)。

MMCV 支持 OpenMMLab 下的眾多計算機視覺相關的開源項目,如 mmdetection(目標檢測工具箱和基準)、mmsegmentation(語義分割工具箱和基準)、mmpose(姿態估計工具箱和基準)、mmaction2(動作理解工具箱和基準)、mmclassification(圖像分類工具箱和基準)等。

一、構建 mmcv

在安裝 mmcv 之前,請確保 PyTorch 已按照?PyTorch 官方安裝指南成功安裝。這可以使用以下命令進行驗證

python -c 'import torch;print(torch.__version__)'

如果輸出版本信息,則安裝 PyTorch。

注意

如果您想使用 代替 , 例如,在最小的容器環境中或沒有 GUI 的服務器中, 您可以在安裝 MMCV 之前先安裝它,以跳過安裝。opencv-python-headlessopencv-pythonopencv-python

  1. 克隆存儲庫

    git clone https://github.com/open-mmlab/mmcv.git
    cd mmcv

    注意

    目前最新的MMSegmentation 需要 MMCV >=2.0.0rc4 但 <2.2.0,推薦安裝?MMCV 2.1.0 而不是最新的?2.2.0 版本,舊版本的?MMCV?從 MMCV 官網的 releases 中下載,也可以從下面的鏈接直接下載。

    MMCV 2.1.0

  2. 安裝并加快編譯速度ninjapsutil

    pip install -r requirements/optional.txt
  3. 檢查 nvcc 版本(需要 9.2+,如果沒有可用的 GPU,請跳過。)

    nvcc --version

    如果上面命令輸出以下消息,則表示nvcc設置正常,否則需要設置CUDA_HOME。

    nvcc: NVIDIA (R) Cuda compiler driver
    Copyright (c) 2005-2020 NVIDIA Corporation
    Built on Mon_Nov_30_19:08:53_PST_2020
    Cuda compilation tools, release 11.2, V11.2.67
    Build cuda_11.2.r11.2/compiler.29373293_0
  4. 檢查 gcc 版本(需要 5.4+)

    gcc --version
  5. 開始構建(需要 10+ 分鐘)

    pip install -e . -v
  6. 驗證安裝

    python .dev_scripts/check_installation.py

    如果上述命令未報告錯誤,則安裝成功。

二、構建 mmcv-lite(可選)

如果您需要使用 PyTorch 相關模塊,請參考?PyTorch 官方安裝指南,確保 PyTorch 已成功安裝到您的環境中。

  1. 克隆存儲庫

    git clone https://github.com/open-mmlab/mmcv.git
    cd mmcv
  2. 開始構建

    MMCV_WITH_OPS=0 pip install -e . -v
  3. 驗證安裝

    python -c 'import mmcv;print(mmcv.__version__)'

    如有其他疑問可以訪問MMCV官網進行查詢

    https://mmcv.readthedocs.io/en/latest/get_started/build.html#https://mmcv.readthedocs.io/en/latest/get_started/build.html#

本文來自互聯網用戶投稿,該文觀點僅代表作者本人,不代表本站立場。本站僅提供信息存儲空間服務,不擁有所有權,不承擔相關法律責任。
如若轉載,請注明出處:http://www.pswp.cn/diannao/98845.shtml
繁體地址,請注明出處:http://hk.pswp.cn/diannao/98845.shtml
英文地址,請注明出處:http://en.pswp.cn/diannao/98845.shtml

如若內容造成侵權/違法違規/事實不符,請聯系多彩編程網進行投訴反饋email:809451989@qq.com,一經查實,立即刪除!

相關文章

Spring Boot 分布式事務常見問題:Seata、XA 與本地消息表對比

一、前言在單體應用中&#xff0c;事務一般由關系型數據庫本身來保證&#xff0c;通過 ACID 特性實現數據一致性。但隨著微服務架構的普及&#xff0c;應用被拆分為多個獨立服務&#xff0c;數據可能分散在不同數據庫、不同存儲引擎中&#xff0c;傳統的單機事務無法再覆蓋。這…

Transporter App 使用全流程詳解:iOS 應用 ipa 上傳工具、 uni-app 應用發布指南

在 iOS 應用開發與發布過程中&#xff0c;Transporter App 是蘋果官方提供的一款上傳工具&#xff0c;專門用于將 ipa 文件 或 應用資源 上傳到 App Store Connect。 與 Xcode 上傳 相比&#xff0c;Transporter 更加穩定&#xff0c;尤其適合大文件上傳&#xff0c;因此在 iOS…

計算機畢業設計 基于Hadoop的B站數據分析可視化系統的設計與實現 Python 大數據畢業設計 Hadoop畢業設計選題【附源碼+文檔報告+安裝調試】

博主介紹&#xff1a;?從事軟件開發10年之余&#xff0c;專注于Java技術領域、Python、大數據、人工智能及數據挖掘、小程序項目開發和Android項目開發等。CSDN、掘金、華為云、InfoQ、阿里云等平臺優質作者? &#x1f345;文末獲取源碼聯系&#x1f345; &#x1f447;&…

使用自定義LLM和Embedding模型部署Vanna:基于RAG的Text-to-SQL生成

使用自定義LLM和Embedding模型部署Vanna&#xff1a;基于RAG的Text-to-SQL生成 說明&#xff1a; 首次發表日期&#xff1a;2024-07-12Vanna Github地址&#xff1a; https://github.com/vanna-ai/vannaVanna官方文檔&#xff1a; https://vanna.ai/ 部署Vanna時我們可以選擇使…

Linux多線程概念

背景知識內存管理OS進行內存管理不是以字節為單位的&#xff0c;而是以內存塊為單位的&#xff0c;默認大小為4kb&#xff1b;系統和磁盤文件進行IO交互的單位是4kb&#xff08;8個扇區&#xff09;&#xff1b;OS對內存管理實質上是對頁框進行管理。頁框&#xff08;Page Fram…

【Problem】動態規劃之跳躍游戲系列

一、跳躍游戲 55. 跳躍游戲 - 力扣&#xff08;LeetCode&#xff09;https://leetcode.cn/problems/jump-game/description/?envTypeproblem-list-v2&envIddynamic-programming class Solution { public:bool canJump(vector<int>& nums) {// 狀態定義&#x…

射頻EVM

EVM&#xff08;Error Vector Magnitude&#xff0c;誤差矢量幅度&#xff09;是衡量無線通信系統中調制質量的重要指標&#xff0c;尤其用于評估信號的調制誤差和系統性能。它通常用來表示傳輸信號與理想信號之間的偏差&#xff0c;特別是在數字通信中。EVM的基本概念&#xf…

Java 更改 Word 文檔中文本顏色

在日常的自動化文檔處理中&#xff0c;我們經常會遇到需要對 Word 文檔內容進行編程修改的需求&#xff0c;其中一項常見且重要的操作就是更改文本的顏色。無論是為了突出重點、統一品牌風格&#xff0c;還是實現動態內容展示&#xff0c;精準地修改文本顏色都是一個核心痛點。…

STM32—SPI協議

文章目錄一、SPI 協議簡介二、硬件電路2.1.SPI的連接2.2.數據的移位2.3.時序基本單元2.3.1.起始條件和終止條件2.3.2.模式 02.3.3.模式 12.3.4.模式 22.3.5.模式 32.4.時序三、軟件實現四、W25Q644.1.簡介4.2.硬件電路4.3.框圖4.4.操作注意事項五、實驗一、SPI 協議簡介 SPI&a…

Qt中的QWebEngineView

第1章 本地目錄結構1.1 自己寫的兩個網頁(html)mermaid.html &#xff08;自己寫的網頁界面&#xff09;WebTest.html (自己寫的網頁界面)qwebchannel.js (Qt下載安裝之后&#xff0c;會在安裝目錄下有這個文件&#xff0c;需要將安裝目錄下的改文件拷貝…

Flutter 應用國際化 (i18n) 與本地化 (l10n) 完整指南

Flutter 國際化 (i18n) 完全指南&#xff1a;從入門到精通 在現代移動應用開發中&#xff0c;支持多語言是觸達全球用戶的基本要求。Flutter 提供了強大且靈活的國際化 (i18n) 和本地化 (l10n) 支持。本文將帶你從零開始&#xff0c;一步步深入掌握在 Flutter 中實現國際化的幾…

計算機視覺與深度學習 | 計算機視覺中線特征提取與匹配算法綜述

文章目錄 一、線特征提取算法原理 1.1 Hough變換及其優化 1.2 LSD算法 1.3 EDLines算法 二、核心數學公式 2.1 直線表示與誤差計算 2.2 LSD算法關鍵公式 三、線特征匹配算法 3.1 LBD描述符 3.2 匹配策略 四、代碼實現 4.1 LSD線段檢測(Python) 4.2 LBD特征匹配(C++) 五、算…

Transformer 模型:Attention is All You Need 的真正含義

2017 年&#xff0c;Google Brain 發布了一篇具有里程碑意義的論文——《Attention Is All You Need》&#xff0c;這篇論文不僅首次提出了 Transformer 模型&#xff0c;更重要的是&#xff0c;它宣稱“注意機制&#xff08;Attention Mechanism&#xff09;就足以構建強大的模…

數據庫約束表的設計

數據庫約束概念&#xff1a;數據庫約束是關系型數據庫的一個重要功能&#xff0c;主要是保證數據的完整性&#xff0c;也可理解為數據的正確性&#xff08;數據本身是否正確&#xff0c;關聯關系是否正確&#xff09;&#xff08;一般是用在指定列上&#xff09;常見的約束類型…

【案例分享】TeeChart 助力 Softdrill 提升油氣鉆井數據可視化能力

在鉆井與地質工程領域&#xff0c;數據可視化是核心環節。圖表不僅需要精確與高效&#xff0c;還需符合行業習慣并支持交互與定制。Softdrill 自 2012 年起在核心產品中集成了TeeChart 圖表庫&#xff0c;將復雜的井下數據轉化為直觀的工程圖表&#xff0c;極大提升了鉆井工程師…

【Flink】Flink Runtime 架構設計

Flink Runtime 架構設計 整體架構 ┌─────────────────────────────────────────────────────────────────┐ │ Flink Runtime │ ├─────────…

Git 命令教程

Git介紹 分布式版本控制系統。 Git命令 初始化/全局配置git init初始化一個Git倉庫&#xff08;會創建一個.git的目錄&#xff09;git config --global user.name “name”設置提交時的用戶名git config user.name查看設置的用戶名git config --global user.email “youemail.c…

git config --global user.name指令報錯時的解決方案

問題分析 %HOMEDRIVE%%HOMEPATH%/.gitconfig 是Windows環境變量的表示方式&#xff1a; %HOMEDRIVE% 通常是 C:%HOMEPATH% 通常是 \Users\你的用戶名完整路徑應該是&#xff1a;C:\Users\你的用戶名\.gitconfig 但這里環境變量沒有被正確解析&#xff0c;顯示的是字面意思。 …

websocket和socket io的區別

好的&#xff0c;這是一個更具體也更常見的問題。WebSocket 是一種協議&#xff0c;而 Socket.IO 是一個庫&#xff0c;它使用了 WebSocket 但提供了多得多的功能。 簡單比喻&#xff1a; WebSocket 就像是給你提供了一條高效的“快遞專線”&#xff08;雙向通信通道&#xff…

Nginx反向代理與負載均衡部署

Nginx反向代理與負載均衡部署實戰指南前言一、規劃部署負載均衡和反向代理二、部署Nginx負載均衡器2.1. 準備基礎環境2.2. 創建Nginx運行用戶2.3. 編譯安裝Nginx2.4. 配置Nginx系統服務2.5. 驗證Nginx安裝三、部署后端2臺Tomcat應用服務器3.1. 安裝JDK3.2. 部署Tomcat實例13.3.…