PyTorch 安装教程
1、查看显卡驱动支持的 CUDA 最高版本
在“显示”一栏中,可以看到“驱动程序版本”为 526.23
找到 NVCUDA.DLL 一栏,产品名称为“NVIDIA CUDA 12. 2.79 driver”,可以看到 CUDA 版
本为 12. 2.79
注意:这里的 12. 2.79 只是表示电脑支持的 CUDA 版本最高为 12. 2.79,不一定非要安装
12. 2.79,可以选择安装 10.2.89、10.1、10.0、9.0 等等,需要根据实际情况。即:可以低
于 12. 2.79,但不能高于它。
2、安装 CUDA
右击下载好的 cuda_10.2.89_441.22_win10.exe 软件,以管理员身份运行,如下图所示:
之后显示窗口
注意:②号红色方框内的路径是安装包解压路径,软件安装完毕后会自动删除。
首先点击①号蓝色框内按钮,选择 CUDA 的解压路径,之后点击③号橘色方框内的 OK 按钮
以继续。
之后将进行解压缩操作,如下图所示:
在解压缩过程中,如果安装 360 等安全软件,会提示软件访问关键位置的警示,选择允许
软件所有操作。
解压完成后,安装软件会自行启动,如下图:
之后软件会进行系统兼容性检查,如下图:
系统兼容性检查通过后,安装程序显示软件许可协议。如下图: