sagemaker OOM
#3
by
formfollows
- opened
deploying this on sagemaker on a g6e.2x gives me a
torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 128.00 MiB. GPU 0 has a total capacity of 44.53 GiB
any setting i'm missing?