smy20011/dreambooth-docker

Update requirements

Closed this issue · 1 comments

Hello again! First, thanks for building this.
When trying to run the container, it says that it is not able to use the GPU.
I checked and I was missing the nvidia-container-toolkit.
Maybe you can add that it is needed to run this correctly?

Thanks!

Thank you! 74b1ff3