ptrblck:

In the latter case CUDA (and other libs) will be shipped in the binaries, so that you don’t have to install a local CUDA version (and it will be ignored).


I installed PyTorch with pip, but I’m not sure if I can use conda in my environment for some reason. Any chance that I could switch the version of cuda to 10 without using conda?
FYI, I found that this problem occurs only if the batchsize on each gpu is over 1. If there is exactly one image on each gpu, nothing would happen.