Zj-BinXia/DiffIR

Motion deblur:Out of memory

Opened this issue · 1 comments

Hi, I try to reproduce your motion deblur training setting by only a single V100 GPU(32GB), but gpu is out of memory. I check previous issues saw that you are training on 8 V100 GPUs, I want to know whether the setting is wrong or not? Looking forward to your reply!
here are my environment:
_libgcc_mutex 0.1 main defaults
_openmp_mutex 5.1 1_gnu defaults
absl-py 1.4.0 pypi_0 pypi
addict 2.4.0 pypi_0 pypi
asttokens 2.2.1 pypi_0 pypi
backcall 0.2.0 pypi_0 pypi
basicsr 1.4.2 pypi_0 pypi
boto3 1.17.112 pypi_0 pypi
botocore 1.20.112 pypi_0 pypi
ca-certificates 2023.01.10 h06a4308_0 defaults
cachetools 5.3.1 pypi_0 pypi
certifi 2023.7.22 pypi_0 pypi
charset-normalizer 3.2.0 pypi_0 pypi
click 8.1.6 pypi_0 pypi
cmake 3.27.0 pypi_0 pypi
contourpy 1.1.0 pypi_0 pypi
cycler 0.11.0 pypi_0 pypi
cython 0.29.30 pypi_0 pypi
debugpy 1.6.7 pypi_0 pypi
decorator 5.1.1 pypi_0 pypi
einops 0.7.0 pypi_0 pypi
entrypoints 0.4 pypi_0 pypi
executing 1.2.0 pypi_0 pypi
facexlib 0.3.0 pypi_0 pypi
filelock 3.12.2 pypi_0 pypi
filterpy 1.4.5 pypi_0 pypi
flask 2.0.1 pypi_0 pypi
fonttools 4.41.1 pypi_0 pypi
future 1.0.0 pypi_0 pypi
fvcore 0.1.5.post20221221 pypi_0 pypi
gfpgan 1.3.8 pypi_0 pypi
google-auth 2.22.0 pypi_0 pypi
google-auth-oauthlib 1.0.0 pypi_0 pypi
grpcio 1.56.2 pypi_0 pypi
gunicorn 20.1.0 pypi_0 pypi
idna 3.4 pypi_0 pypi
imageio 2.34.0 pypi_0 pypi
importlib-metadata 6.8.0 pypi_0 pypi
iopath 0.1.10 pypi_0 pypi
ipykernel 6.7.0 pypi_0 pypi
ipython 8.14.0 pypi_0 pypi
itsdangerous 2.1.2 pypi_0 pypi
jedi 0.18.2 pypi_0 pypi
jinja2 3.0.1 pypi_0 pypi
jmespath 0.10.0 pypi_0 pypi
joblib 1.3.1 pypi_0 pypi
jupyter-client 7.4.9 pypi_0 pypi
jupyter-core 5.3.0 pypi_0 pypi
kiwisolver 1.4.4 pypi_0 pypi
lazy-import 0.2.2 pypi_0 pypi
lazy-loader 0.3 pypi_0 pypi
ld_impl_linux-64 2.38 h1181459_1 defaults
libffi 3.3 he6710b0_2 defaults
libgcc-ng 11.2.0 h1234567_1 defaults
libgomp 11.2.0 h1234567_1 defaults
libstdcxx-ng 11.2.0 h1234567_1 defaults
lit 16.0.6 pypi_0 pypi
llvmlite 0.42.0 pypi_0 pypi
lmdb 1.4.1 pypi_0 pypi
lxml 4.9.1 pypi_0 pypi
ma-tensorboard 1.0.0 pypi_0 pypi
markdown 3.4.4 pypi_0 pypi
markupsafe 2.1.3 pypi_0 pypi
matplotlib 3.6.1 pypi_0 pypi
matplotlib-inline 0.1.6 pypi_0 pypi
mmcv 1.7.1 pypi_0 pypi
mpmath 1.3.0 pypi_0 pypi
natsort 8.4.0 pypi_0 pypi
ncurses 6.4 h6a678d5_0 defaults
nest-asyncio 1.5.6 pypi_0 pypi
networkx 2.6.3 pypi_0 pypi
ninja 1.11.1.1 pypi_0 pypi
numba 0.59.1 pypi_0 pypi
numpy 1.21.5 pypi_0 pypi
nvidia-cublas-cu11 11.10.3.66 pypi_0 pypi
nvidia-cuda-nvrtc-cu11 11.7.99 pypi_0 pypi
nvidia-cuda-runtime-cu11 11.7.99 pypi_0 pypi
nvidia-cudnn-cu11 8.5.0.96 pypi_0 pypi
oauthlib 3.2.2 pypi_0 pypi
opencv-python 4.7.0.72 pypi_0 pypi
openssl 1.1.1t h7f8727e_0 defaults
packaging 23.1 pypi_0 pypi
pandas 1.5.1 pypi_0 pypi
parso 0.8.3 pypi_0 pypi
pathlib2 2.3.7.post1 pypi_0 pypi
pexpect 4.8.0 pypi_0 pypi
pickleshare 0.7.5 pypi_0 pypi
pillow 9.3.0 pypi_0 pypi
pip 22.0.4 pypi_0 pypi
platformdirs 3.5.1 pypi_0 pypi
portalocker 2.8.2 pypi_0 pypi
prompt-toolkit 3.0.38 pypi_0 pypi
protobuf 3.20.1 pypi_0 pypi
psutil 5.9.4 pypi_0 pypi
ptflops 0.7.2.2 pypi_0 pypi
ptyprocess 0.7.0 pypi_0 pypi
pure-eval 0.2.2 pypi_0 pypi
pyasn1 0.5.0 pypi_0 pypi
pyasn1-modules 0.3.0 pypi_0 pypi
pygments 2.15.1 pypi_0 pypi
pyparsing 3.1.0 pypi_0 pypi
pytest-runner 5.3.0 pypi_0 pypi
python 3.9.11 h12debd9_2 defaults
python-dateutil 2.8.2 pypi_0 pypi
pytz 2023.3 pypi_0 pypi
pyyaml 6.0 pypi_0 pypi
pyzmq 25.1.0 pypi_0 pypi
readline 8.2 h5eee18b_0 defaults
realesrgan 0.2.5.0 dev_0
requests 2.28.2 pypi_0 pypi
requests-oauthlib 1.3.1 pypi_0 pypi
rsa 4.9 pypi_0 pypi
s3transfer 0.4.2 pypi_0 pypi
scikit-image 0.22.0 pypi_0 pypi
scikit-learn 1.1.3 pypi_0 pypi
scipy 1.7.3 pypi_0 pypi
setuptools 67.8.0 py39h06a4308_0 defaults
six 1.16.0 pypi_0 pypi
sqlite 3.41.2 h5eee18b_0 defaults
stack-data 0.6.2 pypi_0 pypi
sympy 1.12 pypi_0 pypi
tb-nightly 2.17.0a20240401 pypi_0 pypi
tensorboard 2.12.2 pypi_0 pypi
tensorboard-data-server 0.7.1 pypi_0 pypi
tensorboard-plugin-wit 1.8.1 pypi_0 pypi
termcolor 2.4.0 pypi_0 pypi
threadpoolctl 3.2.0 pypi_0 pypi
tifffile 2024.2.12 pypi_0 pypi
tk 8.6.12 h1ccaba5_0 defaults
tomli 2.0.1 pypi_0 pypi
torch 1.13.1 pypi_0 pypi
torchaudio 0.13.1 pypi_0 pypi
torchvision 0.14.1 pypi_0 pypi
tornado 6.3.2 pypi_0 pypi
tqdm 4.65.0 pypi_0 pypi
traitlets 5.9.0 pypi_0 pypi
triton 2.0.0 pypi_0 pypi
typing-extensions 4.6.3 pypi_0 pypi
tzdata 2023c h04d1e81_0 defaults
urllib3 1.26.16 pypi_0 pypi
wcwidth 0.2.6 pypi_0 pypi
werkzeug 2.3.6 pypi_0 pypi
wheel 0.38.4 py39h06a4308_0 defaults
xz 5.4.2 h5eee18b_0 defaults
yacs 0.1.8 pypi_0 pypi
yapf 0.40.1 pypi_0 pypi
zipp 3.16.2 pypi_0 pypi
zlib 1.2.13 h5eee18b_0 defaults

I have not encountered any problems. You can reduce the number of images in the batch