speedinghzl / CCNet

CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).
MIT License
1.42k stars 277 forks source link

RuntimeError: CUDA Error encountered in <function CompiledLib.bn_mean_var_cuda at 0x7f64a7e57d08> #39

Open jwnirvana opened 5 years ago

jwnirvana commented 5 years ago

when ./libs/build.py is excuted: cc1: warning: command line option ‘-std=c++11’ is valid for C++/ObjC++ but not for C gcc -pthread -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -I/home/wj/anaconda3/envs/torch0.4.0/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include -I/home/wj/anaconda3/envs/torch0.4.0/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH -I/home/wj/anaconda3/envs/torch0.4.0/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/THC -I/usr/local/cuda/include -I/home/wj/anaconda3/envs/torch0.4.0/include/python3.6m -c /media/wj/bangong/wang/CCNet-master/libs/src/lib_cffi.cpp -o ./media/wj/bangong/wang/CCNet-master/libs/src/lib_cffi.o -std=c++11 cc1plus: warning: command line option ‘-Wstrict-prototypes’ is valid for C/ObjC but not for C++ g++ -pthread -shared -L/home/wj/anaconda3/envs/torch0.4.0/lib -Wl,-rpath=/home/wj/anaconda3/envs/torch0.4.0/lib,--no-as-needed ./__ext.o ./media/wj/bangong/wang/CCNet-master/libs/src/lib_cffi.o /media/wj/bangong/wang/CCNet-master/libs/src/bn.o -L/home/wj/anaconda3/envs/torch0.4.0/lib -lpython3.6m -o ./__ext.so


when ./cc_attention/bulid.py is excuted: cc1plus: warning: command line option ‘-Wstrict-prototypes’ is valid for C/ObjC but not for C++ cc1plus: warning: command line option ‘-std=c99’ is valid for C/ObjC but not for C++ g++ -pthread -shared -L/home/wj/anaconda3/envs/torch0.4.0/lib -Wl,-rpath=/home/wj/anaconda3/envs/torch0.4.0/lib,--no-as-needed ./__ext.o ./media/wj/bangong/wang/CCNet-master/cc_attention/src/lib_cffi.o /media/wj/bangong/wang/CCNet-master/cc_attention/src/ca.o -L/home/wj/anaconda3/envs/torch0.4.0/lib -lpython3.6m -o ./__ext.so

ZshahRA commented 4 years ago

Hello All,

Could you please explain why it gives the error below.

python train.py --data-dir ${YOUR_CS_PATH} --random-mirror --random-scale --restore-from ./dataset/resnet101-imagenet.pth --gpu 0,1,2,3 --learning-rate 1e-2 --input-size 769,769 --weight-decay 1e-4 --batch-size 8 --num-steps 60000 --recurrence 2