I am installing ‘pytorch’ on the computer where ‘titan v’ is installed.
This is what i used.

Linux ubuntu 18.04
pytorch 0.4.1
cuda 9.2

And I confirmed that the basic example code (gpu version) provided by ‘pytorch’ works well.
But when i run my code, the following error has occurred.
# RuntimeError "sizes must be non-negative"
My code works fine in “cuda 8.0, torch 0.4.0”.
What should I do?
Should i need to change version for pytorch or cuda?
Or win10 is better to use titanv then linux…? (I’m comfortable with ‘win10’)