Open blueroutecn opened 6 months ago
torch.cuda.OutOfMemoryError: Allocation on device 0 would exceed allowed memory. (out of memory) Currently allocated : 1.14 GiB Requested : 28.12 MiB Device limit : 23.69 GiB Free (according to CUDA): 8.94 MiB PyTorch limit (set by user-supplied memory fraction) : 17179869184.00 GiB
我用两张3090,启动命令: CUDA_VISIBLE_DEVICES=0,1 python main.py
请教如何解决内存不足的问题?
软件都升级了。
torch.cuda.OutOfMemoryError: Allocation on device 0 would exceed allowed memory. (out of memory) Currently allocated : 1.14 GiB Requested : 28.12 MiB Device limit : 23.69 GiB Free (according to CUDA): 8.94 MiB PyTorch limit (set by user-supplied memory fraction) : 17179869184.00 GiB
我用两张3090,启动命令: CUDA_VISIBLE_DEVICES=0,1 python main.py
请教如何解决内存不足的问题?