swz30 / Restormer

[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

About training, NCCL

jjjjzyyyyyy opened this issue · comments

RuntimeError: NCCL error in: /opt/conda/conda-bld/pytorch_1616554786529/work/torch/lib/c10d/ProcessGroupNCCL.cpp:33, unhandled cuda error, NCCL version 2.7.8
ncclUnhandledCudaError: Call to CUDA function failed.

How can i fix it?????
Plz help!

Make sure you can follow the steps in https://github.com/swz30/Restormer/blob/main/INSTALL.md
In case the message still persists, make sure that your machine has active GPUs for CUDA installation.