why should allocate so many memory

ilya-shenbin / RecVAE

The official PyTorch implementation of the paper "RecVAE: A New Variational Autoencoder for Top-N Recommendations with Implicit Feedback"

Apache License 2.0

106 stars 28 forks source link

why should allocate so many memory #5

Open skriser opened 2 years ago

skriser commented 2 years ago

i have dataset: train.csv 5869632 number, user:53424,item:10000; then run error; RuntimeError: CUDA out of memory. Tried to allocate 13.12 GiB (GPU 0; 23.70 GiB total capacity; 0 bytes already allocated; 12.33 GiB free; 0 bytes reserved in total by PyTorch)