Pytorch0.4.1 doesn't support cuda9.1?

1

Hi, I’m trying to install pytorch 0.4.1 on ubuntu, and the cuda version is 9.1.85,.
But i can‘t find the pytorch version.
If i install pytorch for cuda9.0 or 9.2, it all get wrong.
9.0
RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:74
9.2
RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at /pytorch/aten/src/THC/THCGeneral.cpp:74

But i need to use a new function in 0.4.1(pos_weight in BCEWithLogitsLoss).
How can i solve it? Thank you!!

[solves]: Sorry, cuda9.0 can use

2

try using docker … thats the easiest way to get around i think (Suggestion Only)

3

go to https://pytorch.org/previous-versions/ , select the version you need ( like cu91/torch-0.4.0-cp27-cp27mu-linux_x86_64.whl), and do pip install.