Do you have to install CUDA10.2 directly or only the PyTorch binaries with the CUDA10.2 runtime?
I’m not sure, if the former use case is even possible, since you are only using a Python notebook and want to install the CUDA lib on the server/vm.
The latter should be possible by using the install instructions 96 in the notebook.