import torch
print(torch.__version__) # '1.7.0'
print(torch.version.cuda) # 10.2
>>> torch.cuda.is_available() # False
nvidia-smi查看本机linux的型号
访问以下网站,查看各个对应的版本号,已经安装方式
https://pytorch.org/get-started/previous-versions/添加链接描述
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html