首先是安装自己的显卡驱动,根据显卡的型号进行选择
https://www.nvidia.cn/Download/index.aspx
查看显卡驱动和cuda的对应版本
Release Notes :: CUDA Toolkit Documentation
查看显卡的计算能力,该计算能力直接与pytorch的版本有关
CUDA GPUs | NVIDIA Developer
关于CUDA与GPU的算力匹配问题
https://www.csdn.net/tags/MtjakgwsNjE4NjAtYmxvZwO0O0OO0O0O.html
已知cuda版本查看和安装所需要的pytorch版本
Start Locally | PyTorch