import sys, os
sys.path.append(os.path.dirname(os.path.abspath(file)) + ‘/Data’)
CLASS_COUNT = 1
BATCH_SIZE = 2
how can i fix it please help?
Is there more source code than that? Nothing in your code interacts with the GPU.
)
THCudaCheck FAIL file=/pytorch/aten/src/THC/THCTensorRandom.cu line=25 error=2 : out of memory
Traceback (most recent call last):
File “train.py”, line 110, in
main()
File “train.py”, line 56, in main
dream_model.cuda()
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 249, in cuda
return self._apply(lambda t: t.cuda(device))
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 176, in _apply
module._apply(fn)
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 176, in _apply
module._apply(fn)
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 176, in _apply
module._apply(fn)
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 182, in _apply
param.data = fn(param.data)
File “/mnt/home/dreamdd/newDream/DreamHU/venv/lib/python2.7/site-packages/torch/nn/modules/module.py”, line 249, in
return self._apply(lambda t: t.cuda(device))
RuntimeError: cuda runtime error (2) : out of memory at /pytorch/aten/src/THC/THCTensorRandom.cu:25
Can you post your full source code? Or a minimum example that you have run that gets the error message? This is not enough information to help you.