Skip to content

GPU memory use on train.py #14

Description

@sxdl

Thanks for you fascinating job!
I'm trying to run the giving train code on 4060ti 16GB GPU. But it ends up with "CUDA out of memory". I wonder how large will the memory be enough to run the train.py code? I noticed that in #13 it was mentioned that it took a little GPU memory. or is there some magic that can optimize the GPU memory use?
Thank you!

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions