zhangyi-3 / UCDIR

A Unified Conditional Framework for Diffusion-based Image Restoration

Home Page:https://zhangyi-3.github.io/project/UCDIR/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Environment setting or command setting?

csZcWu opened this issue · comments

Hi, thanks for sharing your great job.

When I tried testing the code with 2 GPUs by modifying the command as

'CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch
--nproc_per_node=2 --master_port=4321
python -u sr.py -p val -c config/sid.yaml
--checkpoint experiments/sid/checkpoint/I_Elatest'

it raised an error as following

'/root/anaconda3/envs/UCDIR/bin/python: can't open file './UCDIR-main/python': [Errno 2] No such file or directory'.

How to fix this error?