安装NVIDA驱动
官网下载
https://www.nvidia.cn/Download/index.aspx?lang=cn
选择合适的版本
下载并安装至c盘 选择默认即可
打开NVIDA 点击帮助 选择系统信息
查看配置
可以选择12.1 版本以下的CUDA 的pytorch
Pytorch 安装
点击https://pytorch.org/
进入安装界面 选择合适的版本
https://www.nvidia.cn/Download/index.aspx?lang=cn
可以选择12.1 版本以下的CUDA 的pytorch
进入安装界面 选择合适的版本