pytorch-sentiment-neuron icon indicating copy to clipboard operation
pytorch-sentiment-neuron copied to clipboard

Cuda library problem

Open akbari59 opened this issue 7 years ago • 0 comments

I am trying to use the repository for training a new model but I got the following error

/Users/virtualenv-dir/py3/bin/python /Users/Documents/Codes1/pytorch-sentiment-neuron-master/convert_to_cpu.py /Users/virtualenv-dir/py3/lib/python3.6/site-packages/torch/serialization.py:325: SourceChangeWarning: source code of class 'torch.nn.modules.linear.Linear' has changed. you can retrieve the original source code by accessing the object's source attribute or set torch.nn.Module.dump_patches = Trueand use the patch tool to revert the changes. warnings.warn(msg, SourceChangeWarning) Traceback (most recent call last): File "/Users/Documents/Codes1/pytorch-sentiment-neuron-master/convert_to_cpu.py", line 18, in <module> checkpoint = torch.load(opt.load_model) File "/Users/virtualenv-dir/py3/lib/python3.6/site-packages/torch/serialization.py", line 267, in load return _load(f, map_location, pickle_module) File "/Users/virtualenv-dir/py3/lib/python3.6/site-packages/torch/serialization.py", line 426, in _load result = unpickler.load() File "/Users/virtualenv-dir/py3/lib/python3.6/site-packages/torch/serialization.py", line 395, in persistent_load data_type(size), location) File "/Users/virtualenv-dir/py3/lib/python3.6/site-packages/torch/serialization.py", line 87, in default_restore_location result = fn(storage, location) File "/Users/virtualenv-dir/py3/lib/python3.6/site-packages/torch/serialization.py", line 69, in _cuda_deserialize return obj.cuda(device) File "/Users/virtualenv-dir/py3/lib/python3.6/site-packages/torch/_utils.py", line 68, in _cuda with torch.cuda.device(device): File "/Users/virtualenv-dir/py3/lib/python3.6/site-packages/torch/cuda/__init__.py", line 218, in __enter__ self.prev_idx = torch._C._cuda_getDevice() AttributeError: module 'torch._C' has no attribute '_cuda_getDevice'

It seems that the code only run with Cuda, could you please give me a hint to continue?

Thanks, Mohammad

akbari59 avatar Feb 05 '18 09:02 akbari59