“RuntimeError: operator torchvision::nms does not exist”問題分析

問題描述

源碼編譯的pytorch 2.3.0,加pip安裝的torchvision 0.18.0左痢,在import torchvision時報錯:


image.png

但使用pip安裝的pytorch 2.3.0仰楚,就沒有問題职员。

pip安裝的pytorch為什么不會出錯

從下面這張圖片中可以看到敦姻,即使是使用pip安裝的pytorch,torchvision::nms也不是一開始就有的崔梗。而是在import torchvision過程中動態(tài)添加進去的。


image.png

那么torchvision::nms是什么時候開始有的垒在?
寫一個python腳本蒜魄,這個腳本中只有下面一行:

import torchvision

在vscode中開啟調(diào)試。
經(jīng)過調(diào)試场躯,可以找到谈为,是在下圖中這一行把torchvison::nms加進來的。
load_library前:


image.png

load_library后:


image.png

(注:即使安裝同樣的軟件環(huán)境:miniconda3 + torch2.3.0 + torchvision0.18.0踢关,我在兩臺臺式機上結(jié)果為True伞鲫,在虛擬機和筆記本上結(jié)果為False,原因未知)签舞。

源碼編譯的pytorch為什么會出錯

現(xiàn)在換成源碼編譯的pytorch榔昔,開啟調(diào)試。


image.png

可以看到load_library出現(xiàn)了異常瘪菌。
在Debugger窗口檢查變量:


image.png

錯誤原因是有未定義符號撒会。
查看這個符號的名字:


image.png

看看libtorch_cpu.so中有沒有這個符號:


image.png

可以看到libtorch_cpu.so中雖然有這個符號,但多了[abi:cxx11]修飾师妙。
那么pip安裝的torch中的符號是什么樣的诵肛?


image.png

可以看到這里的符號是不帶[abi:cxx11]后綴的。
所以出錯原因找到了,是因為源碼編譯pytorch的時候開啟了_GLIBCXX_USE_CXX11_ABI宏(這個宏默認是開啟的)怔檩。

解決方法

編譯時關(guān)閉_GLIBCXX_USE_CXX11_ABI宏

export GLIBCXX_USE_CXX11_ABI=0
python3 setup.py develop

注意環(huán)境變量前不帶下劃線”_“褪秀。
如果你的libtorch還依賴了其他源碼編譯的c++庫,那這個庫也要關(guān)掉_GLIBCXX_USE_CXX11_ABI宏薛训。但若是C庫或者所有的接口都是C接口媒吗,那就不受影響。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末乙埃,一起剝皮案震驚了整個濱河市闸英,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌介袜,老刑警劉巖甫何,帶你破解...
    沈念sama閱讀 211,348評論 6 491
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異遇伞,居然都是意外死亡辙喂,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,122評論 2 385
  • 文/潘曉璐 我一進店門鸠珠,熙熙樓的掌柜王于貴愁眉苦臉地迎上來巍耗,“玉大人,你說我怎么就攤上這事渐排∩纸酰” “怎么了?”我有些...
    開封第一講書人閱讀 156,936評論 0 347
  • 文/不壞的土叔 我叫張陵飞盆,是天一觀的道長娄琉。 經(jīng)常有香客問我,道長吓歇,這世上最難降的妖魔是什么孽水? 我笑而不...
    開封第一講書人閱讀 56,427評論 1 283
  • 正文 為了忘掉前任,我火速辦了婚禮城看,結(jié)果婚禮上女气,老公的妹妹穿的比我還像新娘险耀。我一直安慰自己吼驶,他們只是感情好,可當我...
    茶點故事閱讀 65,467評論 6 385
  • 文/花漫 我一把揭開白布券躁。 她就那樣靜靜地躺著轰胁,像睡著了一般谒主。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上赃阀,一...
    開封第一講書人閱讀 49,785評論 1 290
  • 那天霎肯,我揣著相機與錄音,去河邊找鬼。 笑死观游,一個胖子當著我的面吹牛搂捧,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播懂缕,決...
    沈念sama閱讀 38,931評論 3 406
  • 文/蒼蘭香墨 我猛地睜開眼允跑,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了搪柑?” 一聲冷哼從身側(cè)響起聋丝,我...
    開封第一講書人閱讀 37,696評論 0 266
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎拌屏,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體术荤,經(jīng)...
    沈念sama閱讀 44,141評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡倚喂,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,483評論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了瓣戚。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片端圈。...
    茶點故事閱讀 38,625評論 1 340
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖子库,靈堂內(nèi)的尸體忽然破棺而出舱权,到底是詐尸還是另有隱情,我是刑警寧澤仑嗅,帶...
    沈念sama閱讀 34,291評論 4 329
  • 正文 年R本政府宣布宴倍,位于F島的核電站,受9級特大地震影響仓技,放射性物質(zhì)發(fā)生泄漏鸵贬。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 39,892評論 3 312
  • 文/蒙蒙 一脖捻、第九天 我趴在偏房一處隱蔽的房頂上張望阔逼。 院中可真熱鬧,春花似錦地沮、人聲如沸嗜浮。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,741評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽危融。三九已至,卻和暖如春雷袋,著一層夾襖步出監(jiān)牢的瞬間专挪,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 31,977評論 1 265
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留寨腔,地道東北人速侈。 一個月前我還...
    沈念sama閱讀 46,324評論 2 360
  • 正文 我出身青樓,卻偏偏與公主長得像迫卢,于是被迫代替她去往敵國和親倚搬。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 43,492評論 2 348

推薦閱讀更多精彩內(nèi)容