Как установить Pytorch для CUDA?

Question

[Timebird]

Требуется установить Pytorch для CUDA.
Сейчас при вызове команд выход таков:

$ cat /usr/local/cuda/version.txt
CUDA Version 10.0.130
$ nvcc --version | grep "release" | awk '{print $6}' | cut -c2-
9.1.85

$ python
>>> import torch
>>> torch.cuda.is_available()
False

Я правильно понимаю, что у меня две разные версии CUDA на компьютере? Как можно адаптировать хотя бы что-нибудь, чтоб pytorch подцепил cuda?
Заранее спасибо!