美文网首页
Yolact+DenseFusion开发记录

Yolact+DenseFusion开发记录

作者: monk1994 | 来源:发表于2019-07-26 15:02 被阅读0次

    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
    

    相关文章

      网友评论

          本文标题:Yolact+DenseFusion开发记录

          本文链接:https://www.haomeiwen.com/subject/lrnlrctx.html