torch1.6使用官网的安装命令完成后进行测试,出现了错误

torch.cuda.get_device_name(0)
torch.cuda.is_available()
torch.cuda.device_count()
torch.cuda.current_device()

python3.6/site-packages/torch/cuda/__init__.py:125: UserWarning:

Tesla T4 with CUDA capability sm_75 is not compatible with the current PyTorch installation.

The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 compute_37.

If you want to use the Tesla T4 GPU with PyTorch, please check the instructions at Start Locally | PyTorch

warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))

'Tesla T4'

解决方式：

pip install --no-cache-dir torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

再次测试：

成功解决

参考：

PyTorch 1.6 - "Tesla T4 with CUDA capability sm_75 is not compatible" - #20 by luiszeni - PyTorch Forums