search

TRI-ML / packnet-sfm

TRI-ML Monocular Depth Estimation Repository
https://tri-ml.github.io/packnet-sfm/
MIT License
1.23k stars 242 forks source link

horovorun Multi gpu Error #165

Closed sjg02122 closed 2 years ago

sjg02122 commented 3 years ago

The model works well on my server

and I try to run the model using multi-gpu (horovorun -np # ~~)

Even though I never change anything, I face the problem GPU out of memory (It was working well only single gpu)

How to can I fix this problem?

VitorGuizilini-TRI commented 3 years ago

Can you try using our provided dockerfile, so it's the same configuration?

sjg02122 commented 3 years ago

I use the conda environments Is it impossible on conda environments?

VitorGuizilini-TRI commented 2 years ago

I guess it is, if you set up all required packages inside the conda environment.