Yolact+DenseFusion开发记录
2019-07-26 本文已影响0人
monk1994
DenseFusion环境配置（有坑，请看完以下内容）：
CUDA10.0+NVIDIA-DRIVER418.43
conda create -n yolact_dense python=3.6.8
conda activate yolact_dense
conda install pillow scipy numpy pyyaml matplotlib
这里安装了最新（2019.7.26）的pytorch_1.1，yolact要求pytorch1.0+
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
DenseFusion的pytorch1.0版本运行是会包如下错误：
module 'lib.knn.knn_pytorch' has no attribute 'knn'
错误原因：import没写对，改变如下所示：
from lib.knn import knn_pytorch >> from lib.knn.knn_pytorch import knn_pytorch
但还会报以下错：
ImportError: /home/kk/PycharmProjects/DenseFusion-Pytorch-1.0/lib/knn/knn_pytorch/knn_pytorch.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN2at19UndefinedTensorImpl10_singletonE
错误原因：github上直接down下来的源码，里面的knn模块是cuda+c语音写的，编译时候环境和我们的不一样。重新编译一下，然后把编译好的文件放在knn目录下
python setup.py build
python setup.py install
如果还报类似错误，（2019.7.26）建议torchvision使用0.2.2版本，在重新编译。
Yolact环境配置：
conda install cython pycocotools matplotlib
conda install -c conda-forge opencv
