Please advise your cuda version

[yangsusanyang]

Thanks very much for your great work!

I want to try your work. I installed pytorch0.4.1 and python 3.6. When I run "python main.py --mode train", got the following error:

++++++++++++++++++++++++ Traceback (most recent call last): File "main.py", line 204, in main(opt) File "main.py", line 178, in main BMN_Train(opt) File "main.py", line 87, in BMN_Train model = torch.nn.DataParallel(model, device_ids=[0, 1]).cuda() File "C:\Anaconda\envs\py0.4.1\lib\site-packages\torch\nn\parallel\data_parallel.py", line 111, in init _check_balance(self.device_ids) File "C:\Anaconda\envs\py0.4.1\lib\site-packages\torch\nn\parallel\data_parallel.py", line 17, in _check_balance dev_props = [torch.cuda.get_device_properties(i) for i in device_ids] File "C:\Anaconda\envs\py0.4.1\lib\site-packages\torch\nn\parallel\data_parallel.py", line 17, in dev_props = [torch.cuda.get_device_properties(i) for i in device_ids] File "C:\Anaconda\envs\py0.4.1\lib\site-packages\torch\cuda__init__.py", line 292, in get_device_properties raise AssertionError("Invalid device id") AssertionError: Invalid device id +++++++++++++++++++++++++++++

I wonder if it is because the cuda version I installed is different from yours. Please advise the cuda version you used.

Thanks