DiffIR icon indicating copy to clipboard operation
DiffIR copied to clipboard

Motion deblur:Out of memory

Open cheng221 opened this issue 10 months ago • 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

cheng221 avatar Apr 02 '24 12:04 cheng221

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

Zj-BinXia avatar Apr 15 '24 04:04 Zj-BinXia