floodsung/LearningToCompare_FSL

There were some errors when I ran this code and hope can get hlep

yesefeel opened this issue · 0 comments

Thank you for your awesome code.
There were some errors when I ran this code and I didn't know what to do.I hope to get some tips from you, thanks.The following is the problem record:

init data folders
init neural networks
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1513363039688/work/torch/lib/THC/generic/THCStorage.cu line=58 error=30 : unknown error
Traceback (most recent call last):
File "/home/jesse/workspace/jisheng/LearningToCompare_FSL-master/omniglot/omniglot_train_one_shot.py", line 257, in
main()
File "/home/jesse/workspace/jisheng/LearningToCompare_FSL-master/omniglot/omniglot_train_one_shot.py", line 133, in main
feature_encoder.cuda(GPU)
File "/home/jesse/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 216, in cuda
return self._apply(lambda t: t.cuda(device))
File "/home/jesse/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 146, in _apply
module._apply(fn)
File "/home/jesse/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 146, in _apply
module._apply(fn)
File "/home/jesse/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 152, in _apply
param.data = fn(param.data)
File "/home/jesse/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 216, in
return self._apply(lambda t: t.cuda(device))
File "/home/jesse/anaconda2/lib/python2.7/site-packages/torch/_utils.py", line 69, in cuda
return new_type(self.size()).copy
(self, async)
File "/home/jesse/anaconda2/lib/python2.7/site-packages/torch/cuda/init.py", line 361, in _lazy_new
return super(_CudaBase, cls).new(cls, *args, **kwargs)
RuntimeError: cuda runtime error (30) : unknown error at /opt/conda/conda-bld/pytorch_1513363039688/work/torch/lib/THC/generic/THCStorage.cu:58

System Configuration: gtx1080ti, cuda8.0, cudnn v5.1
Torch and Torchvision installed by the conda command
Torch:0.3.0
Torchvision:0.2.0
error.txt
thanks very much