Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

out of memory #7

Open
rnunziata opened this issue Jul 22, 2017 · 5 comments
Open

out of memory #7

rnunziata opened this issue Jul 22, 2017 · 5 comments

Comments

@rnunziata
Copy link

about the 44 frame processed ... run out of memory....running
python run_tracker.py -s DragonBaby -d
Any idea why this is happening?

    return torch.cat(inputs, self.dim)
RuntimeError: cuda runtime error (2) : out of memory at /b/wheel/pytorch-src/torch/lib/THC/generic/THCStorage.cu:66

@Adeel-Abdullah
Copy link

I've been getting the exact same error, I was using the nvidia gtx 960M of my laptop which has 2gb of memory, looks like 2gb is simply not enough for this. It'd be nice to know how much gpu memory we do require for running MDNet

@ahkarami
Copy link

I have also tested the code via 4 Gb GPU, but I got the below error:

File "../modules/model.py", line 40, in forward
    torch.cat((pad,pad,pad,pad,x_sq),2)),1)
RuntimeError: cuda runtime error (2) : out of memory at /opt/conda/conda-bld/pytorch_1512386481460/work/torch/lib/THC/generic/THCStorage.cu:58

@ahkarami
Copy link

Hi,
You can change parameter opts['batch_test'] = 256 in [options.py] file, and modify it into (for example) opts['batch_test'] = 25 to run the code via ~ 2GB GPU.

@ahkarami
Copy link

ahkarami commented Apr 18, 2018

the default parameters needs ~ 5.5 Gb GPU memory.

@ciokan
Copy link

ciokan commented Jun 18, 2018

python run_tracker.py -s DragonBaby -d -f
THCudaCheck FAIL file=/pytorch/aten/src/THC/generic/THCStorage.cu line=58 error=2 : out of memory
Traceback (most recent call last):
  File "run_tracker.py", line 329, in <module>
    result, result_bb, fps = run_mdnet(img_list, init_bbox, gt=gt, savefig_dir=savefig_dir, display=display)
  File "run_tracker.py", line 146, in run_mdnet
    bbreg_feats = forward_samples(model, image, bbreg_examples)
  File "run_tracker.py", line 33, in forward_samples
    regions = regions.cuda()
RuntimeError: cuda runtime error (2) : out of memory at /pytorch/aten/src/THC/generic/THCStorage.cu:58

Titan X 12GB - the vcard goes to 20% on memory then this error pops out. Any ideas?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants