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