nqanh / affordance-net

AffordanceNet - Multiclass Instance Segmentation Framework - ICRA 2018
Other
121 stars 54 forks source link

I have a question of installing AffordanceNet with docker #26

Open kape67 opened 6 years ago

kape67 commented 6 years ago

Hi.

I want to use AffordanceNet using docker.

I downloaded docker-caffe and did git clone AffordanceNet.

Docker-caffe have been installed caffe, so I did not anything. ( Make, build and so on )

And cd AffordanceNet/caffe-affordance-net and edit Makefile.config.

uncomment USE_CUDNN := 1 WITH_PYTHON_LAYER := 1

but I got this error

./include/caffe/util/cudnn.hpp: In function 'void caffe::cudnn::createPoolingDesc(cudnnPoolingStruct*, caffe::PoolingParameter_PoolMethod, cudnnPoolingMode_t, int, int, int, int, int, int)': ./include/caffe/util/cudnn.hpp:127:41: error: too few arguments to function 'cudnnStatus_t cudnnSetPooling2dDescriptor(cudnnPoolingDescriptor_t, cudnnPoolingMode_t, cudnnNanPropagation_t, int, int, int, int, int, int)' pad_h, pad_w, stride_h, stride_w)); ^ ./include/caffe/util/cudnn.hpp:15:28: note: in definition of macro 'CUDNN_CHECK' cudnnStatus_t status = condition; \ ^

Can you help me?

I don't know why it had the error.

I did cudnn v5.1, cudnn v6, not use cudnn, but I got the same error.

I use CUDA 8.0.

Thank you.

nqanh commented 6 years ago

Hi, AffordanceNet code includes some custom layers for Caffe. If you use the original Caffe that comes with Docker, you don't have these layers. Therefore, it would be more convenient if you can build everything from source as in the Readme file.