简而言之，先看你的显卡，打开CMD，输入nvidia-smi，右上角有一个CUDA:XX.X表示当前显卡及当前驱动支持的最高版本CUDA。输入nvidia-smi-q可以看到显卡架构（或者直接去Nvidia官网找你的显卡）。再打开这个连接，查看你显卡架构支持的最低版本CUDA。从中选择一个cuda版本torch是自带了cu