NRCan / geo-deep-learning

Deep learning applied to georeferenced datasets
https://geo-deep-learning.readthedocs.io/en/latest/
MIT License
150 stars 49 forks source link

BUG: broken docker image potentially due to memory leak #534

Closed remtav closed 1 year ago

remtav commented 1 year ago

Describe the bug The latest docker image of GDL cannot be used: image

This error is raised on HPC (Apptainer) as well as locally with docker image.

To Reproduce Steps to reproduce the behavior:

docker pull geobaserepo/gdl-cuda11:v2.7.1
docker run -it --rm geobaserepo/gdl-cuda11:v2.7.1 /bin/bash -c 'python /home/gdl_user/geo-deep-learning/GDL.py mode=inference'

Expected behavior This particular command should raise a FileNotFound error, not "std::bad_alloc"

Additional context After some testing, changes made to the environment.yml file seem to be the cause of this issue. In the current version, more libraries are installed via pip, which could cause the conda environment to break. The previous version of this file still works.