yeah… You are right… Check this answer:
You could install a previous version of PyTorch that supports CUDA-9.0…