Running pytroch v. 0.3.1 on cuda 9.0

can i run pytorch v.0.3.1 on cuda 9.0 on windows 10 ?. Actually, in the site i found, it is supposed to be installed with cuda 8.0. but, i just installed cuda 9.0 on my laptop.


The current version is 0.4.0. You can use the instructions on the website to install it for cuda 9.0.