Just a heads-up.
I am not sure if pytorch is already supported with CUDA10. I tried it once recently and some stuffs did not work.
https://discuss.pytorch.org/t/solved-weird-behavior-with-cuda-toolkit-version-10-0-gcc-6-4-0/26870/3 143
Why not try with CUDA9 / CUDA9.2?