These commands should install the stable version with CUDA10:
pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-linux_x86_64.whl
pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp37-cp37m-linux_x86_64.whl




 YN-Tang:

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.


CUDA9 works in some cases with Turing GPUs, but is generally not recommended, so we should make sure you can properly install CUDA10.
Let me know, if these commands worked for you.