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
网友评论