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