search

csxmli2016 / DFDNet

Blind Face Restoration via Deep Multi-scale Component Dictionaries (ECCV 2020)
915 stars 215 forks source link

Provide requirements #1

Closed detkov closed 4 years ago

detkov commented 4 years ago

Hi! Thanks for open-sourcing such a perspective work!

I'm trying to test it in Colab and having some issues with CUDA on the last step (test_FaceDict.py). Can you please provide us with some environment and requirements that can be used to reevaluate results?

Error:

----------------- Options ---------------
<Some options>
----------------- End -------------------
THCudaCheck FAIL file=/pytorch/torch/csrc/cuda/Module.cpp line=59 error=101 : invalid device ordinal
Traceback (most recent call last):
  File "test_FaceDict.py", line 88, in <module>
    opt = TestOptions().parse()
  File "/content/DFDNet/options/base_options.py", line 114, in parse
    torch.cuda.set_device(opt.gpu_ids[0])
  File "/usr/local/lib/python3.6/dist-packages/torch/cuda/__init__.py", line 281, in set_device
    torch._C._cuda_setDevice(device)
RuntimeError: cuda runtime error (101) : invalid device ordinal at /pytorch/torch/csrc/cuda/Module.cpp:59
csxmli2016 commented 4 years ago

Hi! Thanks for open-sourcing such a perspective work!

I'm trying to test it in Colab and having some issues with CUDA on the last step (test_FaceDict.py). Can you please provide us with some environment and requirements that can be used to reevaluate results?

Error:

----------------- Options ---------------
<Some options>
----------------- End -------------------
THCudaCheck FAIL file=/pytorch/torch/csrc/cuda/Module.cpp line=59 error=101 : invalid device ordinal
Traceback (most recent call last):
  File "test_FaceDict.py", line 88, in <module>
    opt = TestOptions().parse()
  File "/content/DFDNet/options/base_options.py", line 114, in parse
    torch.cuda.set_device(opt.gpu_ids[0])
  File "/usr/local/lib/python3.6/dist-packages/torch/cuda/__init__.py", line 281, in set_device
    torch._C._cuda_setDevice(device)
RuntimeError: cuda runtime error (101) : invalid device ordinal at /pytorch/torch/csrc/cuda/Module.cpp:59

Hi, thanks for your suggestions. I found the problem is common in different environments. You can refer to here.

I use pytorch1.3 in this work and it runs successfully.

detkov commented 4 years ago

Thanks, but what CUDA, cuDNN, torchvision and numpy versions do you use?

csxmli2016 commented 4 years ago

Thanks, but what CUDA, cuDNN, torchvision and numpy versions do you use?

Hi, I have updated the pytorch version to 1.5. The new environment is shown below.