[Bug]: Trying to enable multi gpu training does not work

openvinotoolkit / anomalib

An anomaly detection library comprising state-of-the-art algorithms and features such as experiment management, hyper-parameter optimization, and edge inference.

https://anomalib.readthedocs.io/en/latest/

Apache License 2.0

3.41k stars 616 forks source link

[Bug]: Trying to enable multi gpu training does not work #2090

Open KarikalStrom opened 1 month ago

KarikalStrom commented 1 month ago

Describe the bug

Setting devices to -1 or listing them is not working and I always get OOM error after the first GPU get full

Dataset

MVTec

Model

PatchCore

Steps to reproduce the behavior

OS information

Expected behavior

Screenshots

No response

Pip/GitHub

pip

What version/branch did you use?

No response

Configuration YAML

Logs

Code of Conduct

[X] I agree to follow this project's Code of Conduct