run test.py error

[CelineZhou03]

cuda runtime error!
When trying to run test.py I get the following message:
(pytorch) zhouyao@wb-i36g:~/crowdcount-cascaded-mtl$ python test.py --gpu 0
Pre-loading the data. This may take a while...
Loaded 100 / 182
Completed laoding 182 files
/home/zhouyao/anaconda3/envs/pytorch/lib/python2.7/site-packages/h5py/init.py:36: FutureWarning: Conversion of the second argumen t of issubdtype from float to np.floating is deprecated. In future, it will be treated as np.float64 == np.dtype(float).type.
from ._conv import register_converters as _register_converters
/home/zhouyao/crowdcount-cascaded-mtl/src/crowd_count.py:25: UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.
density_cls_prob = F.softmax(density_cls_score)
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1513363039688/work/torch/lib/THC/generic/THCTensorCopy.c line=70 error=77 : an ill egal memory access was encountered
Traceback (most recent call last):
File "test.py", line 45, in
density_map = density_map.data.cpu().numpy()
File "/home/zhouyao/anaconda3/envs/pytorch/lib/python2.7/site-packages/torch/tensor.py", line 35, in cpu
return self.type(getattr(torch, self.class.name))
File "/home/zhouyao/anaconda3/envs/pytorch/lib/python2.7/site-packages/torch/cuda/init.py", line 370, in type
return super(_CudaBase, self).type(*args, **kwargs)
File "/home/zhouyao/anaconda3/envs/pytorch/lib/python2.7/site-packages/torch/_utils.py", line 38, in type
return new_type(self.size()).copy(self, async)
RuntimeError: cuda runtime error (77) : an illegal memory access was encountered at /opt/conda/conda-bld/pytorch_1513363039688/work/t orch/lib/THC/generic/THCTensorCopy.c:70

Any suggestions for what might be wrong.