How to setup deform conv with cuda9.0

xinntao / EDVR

Winning Solution in NTIRE19 Challenges on Video Restoration and Enhancement (CVPR19 Workshops) - Video Restoration with Enhanced Deformable Convolutional Networks. EDVR has been merged into BasicSR and this repo is a mirror of BasicSR.

https://github.com/xinntao/BasicSR

1.49k stars 318 forks source link

How to setup deform conv with cuda9.0 #136

Open x423xu opened 4 years ago

x423xu commented 4 years ago

When I setup deform conv with cuda9.0, it comes out that "ImportError: libcudart.so.10.0: cannot open shared object file: No such file or directory". Please let me know how to solve it.Thanks. Note: it seems that I nned to modify some parameters to make it compiled with CUDA9.0 rather than CUDA10.0. But I can't figure out what to do.

xinntao commented 4 years ago

Try to modify the make.sh file: modify the following paths according to your environments.

CUDA_HOME=/usr/local/cuda-10.0 \
CUDNN_INCLUDE_DIR=/usr/local/cuda-10.0/include \
CUDNN_LIB_DIR=/usr/local/cuda-10.0/lib64 \
python setup.py build develop