I’m not sure but it looks like from this open issue 378 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)