报错
RuntimeError: MPS backend out of memory (MPS allocated: 4.61 GB, other allocations: 8.10 GB, max allowed: 13.57 GB). Tried to allocate 1024.00 MB on private pool. Use PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0 to disable upper limit for memory allocations (may cause system failure).
Time taken: 3.1 sec.
这是由于内存不足导致的,我们设置下环境变量忽略它
export PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0
评论(0)