我現在在跑這個github上的這個程式：https://github.com/rowanz/neural-motifs
我已經在自己的linux服務器上安裝了cude9.0（實驗室的服務器原配時10.1的），除了pytroch0.3.0版本實在安裝不了外，其他所有的東西都是按照這上面的程序配置的
并且在服務器上nvcc -V得出的結果也是nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176
但執行pretrain_detector.sh時仍然會出現這樣的錯誤：
Traceback (most recent call last):
File "/home/lijiayi/neural-motifs-master/models/train_detector.py", line 11, in <module>
from lib.object_detector import ObjectDetector
File "/home/lijiayi/neural-motifs-master/lib/object_detector.py", line 11, in <module>
from lib.fpn.nms.functions.nms import apply_nms
File "/home/lijiayi/neural-motifs-master/lib/fpn/nms/functions/nms.py", line 4, in <module>
from .._ext import nms
File "/home/lijiayi/neural-motifs-master/lib/fpn/nms/_ext/nms/__init__.py", line 3, in <module>
from ._nms import lib as _lib, ffi as _ffi
ImportError: /home/lijiayi/neural-motifs-master/lib/fpn/nms/_ext/nms/_nms.so: undefined symbol: __cudaRegisterFatBinaryEnd
請問有人知道原因嗎
uj5u.com熱心網友回復：
我剛遇到，我裝的cuda是10.1版本的，而Pytorch是0.4.1版本的，兩者不匹配，可以參考https://www.cnblogs.com/luruiyuan/p/10720581.html，不知道你現在解決了沒有～轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/120493.html
上一篇：python串列推導式求助