waifu2x icon indicating copy to clipboard operation
waifu2x copied to clipboard

Use Shared GPU Memory instead of Dedicated GPU Memory

Open GovetaXV opened this issue 6 years ago • 3 comments

Hi! I wanted to ask if there was a way to utilize the shared GPU memory instead of the dedicated one. Even though it is a little slower, There is a lot more of it.

GovetaXV avatar Jan 11 '19 19:01 GovetaXV

(I may not understand what are you asking about.) I do not know whether there is shared memory for GPUs. In training process, you can use NVIDIA NCCL (NVIDIA Collective Communications Library) for multi-GPUs, if you installed NVIDIA NCCL and my modified version of nccl.torch.

nagadomi avatar Jan 12 '19 00:01 nagadomi

If you got GPU out of memory error, you can avoid it with -crop_size option (e.g. -crop_size 128).

nagadomi avatar Jan 12 '19 00:01 nagadomi

It is the shared memory windows allocates to a gpu in the event you run out of VRAM during a game. In gaming the driver handles this by dumping VRAM contents into RAM. CUDA supports this with shared memory, or unified memory, something like that, but it requires explicit programming to do so.

78Alpha avatar Jul 26 '20 21:07 78Alpha