OS: Ubuntu 16.04 LTS
GPU: Nvidia k80 12gb memory
Cuda: v9
conda 4.5.1
Python: 3.6
Pytorch: 0.4

After I

1. download model
2. pip installed python_speech_features

Ran:

python testSyncNet.py --video data/example.avi

Error:

THCudaCheck FAIL file=torch/csrc/cuda/Module.cpp line=32 error=10 : invalid device ordinal
Traceback (most recent call last):
File "testSyncNet.py", line 164, in
s.loadParameters(args.initial_model);
File "testSyncNet.py", line 150, in loadParameters
loaded_state = torch.load(path);
File "/home/T/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 303, in load
return _load(f, map_location, pickle_module)
File "/home/T/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 469, in _load
result = unpickler.load()
File "/home/T/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 437, in persistent_load
data_type(size), location)
File "/home/T/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 88, in default_restore_location
result = fn(storage, location)
File "/home/T/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 70, in _cuda_deserialize
return obj.cuda(device)
File "/home/T/anaconda3/lib/python3.6/site-packages/torch/_utils.py", line 68, in _cuda
with torch.cuda.device(device):
File "/home/T/anaconda3/lib/python3.6/site-packages/torch/cuda/init.py", line 227, in enter
torch._C._cuda_setDevice(self.idx)
RuntimeError: cuda runtime error (10) : invalid device ordinal at torch/csrc/cuda/Module.cpp:32

Note.. i can run TF / keras just fine on this machine..