Tracy Teal
04/23/2024, 7:14 PMCristian Challu
04/23/2024, 7:19 PMCristian Challu
04/23/2024, 7:20 PMTracy Teal
04/23/2024, 7:23 PMMarco
04/23/2024, 7:57 PMMarco
04/23/2024, 8:17 PMMarco
04/23/2024, 8:50 PMRuntimeError: MPS backend out of memory (MPS allocated: 0 bytes, other allocations: 0 bytes, max allowed: 7.93 GB). Tried to allocate 256 bytes on shared pool. Use PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0 to disable upper limit for memory allocations (may cause system failure).
I'm not sure it's a good idea to set an env variable in a notebook. Otherwise, I'm not sure how to fix this