1、环境配置

conda create -n diffir python=3.9
conda activate diffir

去模糊环境:

sudo apt-get update
sudo apt install tmux
sudo apt install libgl1-mesa-glx
pip install basicsr 
pip install -r requirements.txt
pip install pandas 
sudo python3 setup.py develop
pip install einops 
pip install natsort 
pip uninstall nvidia_cublas_cu11
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118

2、train

python DiffIR/train.py -opt options/train_DiffIRS1.yml

运行成功:

3、test

sh test.sh

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐