Cannot select the GPU index

[v-iashin]

Hi.

It seems that I cannot make the code to run on another GPU rather than 0-th. The following line doesn't make any difference in my case

pytorch-pwc/run.py

Line 25 in f613890

torch.cuda.device(1) # change this if you have a multiple graphics cards and you want to utilize them

I managed to solve it by introducing the device variable instead and adding torch.cuda.set_device(device) right after the FunctionCorrelation signature

pytorch-pwc/correlation/correlation.py

Lines 392 to 393 in f613890

	def FunctionCorrelation(tensorFirst, tensorSecond):
	return _FunctionCorrelation.apply(tensorFirst, tensorSecond)

and also passing this variable as an argument here

pytorch-pwc/run.py

Line 196 in f613890

tensorVolume = torch.nn.functional.leaky_relu(input=correlation.FunctionCorrelation(tensorFirst=tensorFirst, tensorSecond=tensorSecond), negative_slope=0.1, inplace=False)

and here

pytorch-pwc/run.py

Line 204 in f613890

tensorVolume = torch.nn.functional.leaky_relu(input=correlation.FunctionCorrelation(tensorFirst=tensorFirst, tensorSecond=Backward(tensorInput=tensorSecond, tensorFlow=tensorFlow * self.dblBackward)), negative_slope=0.1, inplace=False)

If this is a bug do you want me to form a Pull Request?

Anyway, thanks a lot for sharing the code. It is really helpful when the official version is on CUDA 8 and PyTorch 0.2!

[sniklaus]

You are correct, it should be set_device instead of just device. Thank you for letting me know! No need to issue a pull request, I just updated it and removed that line altogether. It is usually better to set CUDA_VISIBLE_DEVICES and I only had this line to avoid issues with multi-GPU systems.