A huge difference of memory usage on different GPU

NVIDIA / pix2pixHD

Synthesizing and manipulating 2048x1024 images with conditional GANs

https://tcwang0509.github.io/pix2pixHD/

Other

6.62k stars 1.39k forks source link

A huge difference of memory usage on different GPU #266

Open wb-08 opened 3 years ago

wb-08 commented 3 years ago

I have two workstation: one with RTX 3090 and another with RTX A6000. The model on RTX 3090 takes 8897MB in total.When it runs on the RTX A6000 it takes 36945MB. I have identical cuda, cudnn and pytorch version

kcshum commented 2 years ago

Same for me, having the RTX 3090 GPU memory usage ~8700MB.