search

joonson / syncnet_python

Out of time: automated lip sync in the wild
MIT License
681 stars 149 forks source link

RuntimeError: cuda runtime error (10) : invalid device ordinal #1

Closed taewookim closed 6 years ago

taewookim commented 6 years ago

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..