New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Out of memory error: Multi-GPUs not used when training semi-supervised model #339

Open

wilson-xu-dev opened this issue Jul 11, 2023 · 0 comments

Assignees

wilson-xu-dev commented Jul 11, 2023

Hi I was trying to follow the "unsupervised_hello_world" notebook tutorial, but trained on Stanford cars dataset.
(https://github.com/tensorflow/similarity/blob/master/examples/unsupervised_hello_world.ipynb).

Running into out of memory error even with 4x 16gb Nvidia V100 as only one gpu is utilised. (Using AWS ec2 instance)

Error message:

nvidia-smi:

Relevant code snippet:

owenvallis self-assigned this

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