PyTorch имеет версию CUDA = 10.2, а torchvision имеет версию CUDA = 9.2. Я установил из souce с
pip install torch==1.5.0 torchvision==0.6.0 -f https://download.pytorch.org/whl/torch_stable.html
, почему он по-прежнему сообщает torchvision с использованием CUDA 9.2