search

JimmyChame / SADNet

Pytorch code for "Spatial-Adaptive Network for Single Image Denoising"
128 stars 12 forks source link

import deform_conv_cuda #15

Closed DivyaSrivastav closed 2 years ago

DivyaSrivastav commented 2 years ago

File "/workspace/divya/Denoising/SADNet/dcn/deform_conv.py", line 10, in from . import deform_conv_cuda ImportError: /workspace/divya/Denoising/SADNet/dcn/deform_conv_cuda.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN3c1019UndefinedTensorImpl10_singletonE

Kindly help

JimmyChame commented 2 years ago

Maybe it's environment version incompatibility. You can configure the environment by following the README.

JimmyChame commented 2 years ago
  1. Make sure Python >= 3.6 Pytorch >= 1.1 CUDA >= 9.0，
  2. Delete build and deform_conv.egg-info folder in /dcn, and recompile dcn according to your environement.
JimmyChame commented 2 years ago

We rebuild our codes based on torchvision >=0.9.0. If torchvision>=0.9.0 (pytorch >= 1.8.0) in your environment, you don't need follow the instructions below to install DCNv2.