Does gpu version of pytorch work with cuda 9.0?

I installed cuda 9.0 so I wanted to know if pytorch works with this version. If not what should i do to run the project with gpu version of pytorch?


Iā€™m not sure but it looks like from this open issue the answer is probably not. You should try to build pytorch from source and see if it works. If not, you might need to install cuda 8.0 (I think the two cudas can live side by side)

I have recently installed PyTorch from source with cuda 9.0. The only thing I needed to do is to remove magma, which was compiled for cuda 8.0. For reference, my environment is Ubuntu 16.04, cuda 9.0, cudNN 7.


Same env as Simon. Everything works out of the box for me with the conda install.

AWS Ubuntui Deep Learning AM cuda 9 I + p3. instances. Installation from source works out of the box and lets you enjoy the new gpus :slight_smile:

Did you use the cuda8 download link from the website, i.e,:conda install pytorch torchvision cuda80 -c soumith

yes, I have used linux - conda - python 3.6 - cuda8.0 on multiple machines.

conda install pytorch torchvision cuda80 -c soumith