1. GPU对应的CUDA版本 nvidia-smi
CUDA Version: 12.2
GPU diver 大于cuda toolkit, pytorch 版本根据cuda toolkit
2. 查看nvcc的版本(即cuda toolkit 版本)
nvcc --version
Cuda compilation tools, release 12.2, V12.2.91
Build cuda_12.2.r12.2/compiler.32965470_0
如未安装,下载对应的cuda版本安装 不要安装驱动
下载对应的cuda
cuda pytorch
cuda 下载Pytorch深度学习实战3-1:最新Windows/Ubuntu双系统Pytorch图文安装教程-CSDN博客