search

archersama / IntTower

Source code of CIKM 2022 and DLP-KDD workshop 2022 Best Paper: IntTower-“ IntTower: the Next Generation of Two-Tower Model for Pre-ranking System”
Apache License 2.0
58 stars 10 forks source link

CUDA out of memory #7

Closed ymh-ymh closed 1 year ago

ymh-ymh commented 1 year ago

RuntimeError: CUDA out of memory. Tried to allocate 32.00 MiB (GPU 0; 23.99 GiB total capacity; 37.27 GiB already allocated; 0 bytes free; 37.33 GiB reserved in total by PyTorch)

I run this code on 24G GPU, this error always happened after epoch 2 whatever batch_size I set, is there anything wrong with my environment?

archersama commented 1 year ago

which python script do you use? I think it has nothing to do with the environment. Maybe more detailed information you can provide

ymh-ymh commented 1 year ago

I take python3.10.12 as my interpreter. should I change the python version?

archersama commented 1 year ago

use this， https://github.com/archersama/IntTower/blob/main/requirements.txt ，and try python3.7