Coder Social home page Coder Social logo

Comments (3)

LiMeng95 avatar LiMeng95 commented on August 31, 2024 1

hi, @LouisNUST

Have you build your own NMS extension? See: prerequisites-NMS extension. And please ensure using the same cuda version with pytorch.

from multiposenet.pytorch.

LouisNUST avatar LouisNUST commented on August 31, 2024

@LiMeng95 Thanks for your reply.
I have precompiled my own NMS extentions:

Compiling nms kernels by nvcc...
Including CUDA code.
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms
generating /tmp/tmpp59_1z1x/nms.c
setting the current directory to '/tmp/tmpp59_1z1x'
running build_ext
building 'nms' extension
creating home
creating home/louis
creating home/louis/Documents
creating home/louis/Documents/DetPose
creating home/louis/Documents/DetPose/MultiPoseNet
creating home/louis/Documents/DetPose/MultiPoseNet/lib
creating home/louis/Documents/DetPose/MultiPoseNet/lib/nms
creating home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src
gcc -pthread -B /home/louis/anaconda3/envs/MultiPose/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -DWITH_CUDA -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/THC -I/usr/local/cuda/include -I/home/louis/anaconda3/envs/MultiPose/include/python3.6m -c nms.c -o ./nms.o -std=c99
gcc -pthread -B /home/louis/anaconda3/envs/MultiPose/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -DWITH_CUDA -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/THC -I/usr/local/cuda/include -I/home/louis/anaconda3/envs/MultiPose/include/python3.6m -c /home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.c -o ./home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.o -std=c99
In file included from /home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/TH.h:4:0,
from /home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.c:1:
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.c: In function ‘cpu_nms’:
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.c:7:42: warning: passing argument 1 of ‘THLongTensor_isContiguous’ from incompatible pointer type [-Wincompatible-pointer-types]
THArgCheck(THLongTensor_isContiguous(boxes), 2, "boxes must be contiguous");
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THGeneral.h:100:35: note: in definition of macro ‘THArgCheck’
THArgCheck(FILE, LINE, VA_ARGS);
^~~~~~~~~~~
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THTensor.h:8:39: note: expected ‘const THLongTensor * {aka const struct THLongTensor *}’ but argument is of type ‘THFloatTensor * {aka struct THFloatTensor *}’
#define THTensor
(NAME) TH_CONCAT_4(TH,Real,Tensor
,NAME)
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THGeneral.h:144:37: note: in definition of macro ‘TH_CONCAT_4_EXPAND’
#define TH_CONCAT_4_EXPAND(x,y,z,w) x ## y ## z ## w
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THTensor.h:8:27: note: in expansion of macro ‘TH_CONCAT_4’
#define THTensor
(NAME) TH_CONCAT_4(TH,Real,Tensor
,NAME)
^~~~~~~~~~~
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/generic/THTensor.h:114:12: note: in expansion of macro ‘THTensor

TH_API int THTensor_(isContiguous)(const THTensor self);
^~~~~~~~~
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.c:9:42: warning: passing argument 1 of ‘THLongTensor_isContiguous’ from incompatible pointer type [-Wincompatible-pointer-types]
THArgCheck(THLongTensor_isContiguous(areas), 4, "areas must be contiguous");
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THGeneral.h:100:35: note: in definition of macro ‘THArgCheck’
THArgCheck(FILE, LINE, VA_ARGS);
^~~~~~~~~~~
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THTensor.h:8:39: note: expected ‘const THLongTensor * {aka const struct THLongTensor *}’ but argument is of type ‘THFloatTensor * {aka struct THFloatTensor *}’
#define THTensor
(NAME) TH_CONCAT_4(TH,Real,Tensor_,NAME)
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THGeneral.h:144:37: note: in definition of macro ‘TH_CONCAT_4_EXPAND’
#define TH_CONCAT_4_EXPAND(x,y,z,w) x ## y ## z ## w
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THTensor.h:8:27: note: in expansion of macro ‘TH_CONCAT_4’
#define THTensor_(NAME) TH_CONCAT_4(TH,Real,Tensor_,NAME)
^~~~~~~~~~~
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/generic/THTensor.h:114:12: note: in expansion of macro ‘THTensor_’
TH_API int THTensor_(isContiguous)(const THTensor self);
^~~~~~~~~
gcc -pthread -B /home/louis/anaconda3/envs/MultiPose/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -DWITH_CUDA -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/THC -I/usr/local/cuda/include -I/home/louis/anaconda3/envs/MultiPose/include/python3.6m -c /home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c -o ./home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.o -std=c99
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c: In function ‘gpu_nms’:
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c:29:35: warning: initialization from incompatible pointer type [-Wincompatible-pointer-types]
unsigned long long
mask_flat = THCudaLongTensor_data(state, mask);
^~~~~~~~~~~~~~~~~~~~~
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c:37:40: warning: initialization from incompatible pointer type [-Wincompatible-pointer-types]
unsigned long long * mask_cpu_flat = THLongTensor_data(mask_cpu);
^~~~~~~~~~~~~~~~~
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c:40:39: warning: initialization from incompatible pointer type [-Wincompatible-pointer-types]
unsigned long long
remv_cpu_flat = THLongTensor_data(remv_cpu);
^~~~~~~~~~~~~~~~~
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c:23:7: warning: unused variable ‘boxes_dim’ [-Wunused-variable]
int boxes_dim = THCudaTensor_size(state, boxes, 1);
^~~~~~~~~
gcc -pthread -shared -B /home/louis/anaconda3/envs/MultiPose/compiler_compat -L/home/louis/anaconda3/envs/MultiPose/lib -Wl,-rpath=/home/louis/anaconda3/envs/MultiPose/lib -Wl,--no-as-needed -Wl,--sysroot=/ ./_nms.o ./home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.o ./home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.o /home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/cuda/nms_kernel.cu.o -o ./_nms.so

the CUDA version is 10.1
Ubuntu: 18.04
python: 3.6.8
pytorch: 0.4.1
cuda90: 1.0

from multiposenet.pytorch.

LiMeng95 avatar LiMeng95 commented on August 31, 2024

@LouisNUST ,

Maybe the mismatch of CUDA version caused this problem. I used CUDA9.0, so that you can find package cuda90: 1.0 in multipose_environment.yaml

from multiposenet.pytorch.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.