Specific versions of dependencies/requirements

[xylhal]

Hi there,

Is it possible to provide the specific versions of the packages required or a requirements.txt to run the experiments?

I have installed the required packages but there seems to be a conflict with the package versions which causes Pytorch not to recognize the GPUs.

[ondrejbohdal]

Thank you for your interest in Meta Omnium and apologies for the delayed response!

We have used the following versions of the packages:

numpy==1.19.2
opencv-python==4.6.0
scikit-learn==0.24.2
torch==1.8.1
scipy==1.6.3
tqdm==4.60.0
optuna==3.1.0

Unrelated to Meta Omnium, we have also recently seen issues with PyTorch not recognizing GPUs after installation in a new conda environment. We have been able to resolve these by running the following twice (somehow it seems to have installed CPU version only when running it for the first time, but repeating it helped):
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia