search

Mukosame / Zooming-Slow-Mo-CVPR-2020

Fast and Accurate One-Stage Space-Time Video Super-Resolution (accepted in CVPR 2020)
GNU General Public License v3.0
915 stars 164 forks source link

error compiling DCN #19

Closed tonyrewin closed 4 years ago

tonyrewin commented 4 years ago 
Zooming-Slow-Mo-CVPR-2020/codes/models/modules/DCNv2/src/cuda/dcn_v2_cuda.cu(279): error: identifier "THCState_getCurrentStream" is undefined

Zooming-Slow-Mo-CVPR-2020/codes/models/modules/DCNv2/src/cuda/dcn_v2_cuda.cu(279): error: identifier "THCState_getCurrentStream" is undefined

2 errors detected in the compilation of "/tmp/tmpxft_00000988_00000000-6_dcn_v2_cuda.cpp1.ii".
error: command '/usr/bin/nvcc' failed with exit status 1

ubuntu 20.04, cuda 10.1, python 3.8

tonyrewin commented 4 years ago

got it fixed with a trick similar to this change: https://github.com/pytorch/pytorch/pull/14500/files

Mukosame commented 4 years ago

Thank you!