在配CVPR2021-Probabilistic Modeling of Semantic Ambiguity for Scene Graph Generation这篇论文的环境时,遇到了这个问题:
环境:矩池云租的GPU TeslaK80,torch0.4.1,cuda90(矩池云直接就有这个镜像妈的我爆哭)
这篇论文在motifs的基础上改的,但是motif已经是18年的论文了,环境一定要是0.4的torch,有个ffi库(好像叫这个)在torch1.0之后就弃用了。
这两个文件夹下都有src/cuda文件夹,cd进去之后有一个makefile文件,在makefile的最开始加上这句:
NVCC=/usr/local/cuda-9.0/bin/nvcc
再cd到roi_align或者nms文件,在终端里输入 make
bug解决
参考链接:https://blog.csdn.net/a362682954/article/details/103539163
----------------------一些碎碎念-------------------------
我从中秋节开始配这个环境。
是的。中秋节。
中秋节三天假我在实验室的电脑配这个环境,死活配不出来。
然后断断续续又配过几次,但都失败了。
这次是必须要配好了。因为老师开始push我了。
一开始用的AutoDL的GPU去配,但是AutoDL它的python只有3.8一种,而我去官网下torch0.4的轮子的时候,他喵的最高只支持3.7!!!一上午的时间没了。
下午开始用矩池云配,矩池云yyds!!!它居然有torch0.4的镜像!!!!!
555我永远爱矩池云QAQ
-------------------碎碎念后的正事--------------------
编译顺序:
.c/.cu文件→.o文件→.so文件
.so文件才是解决bug的关键。