网上资源越来越多,关于PyTorch的安装教程各式各样,下面我将详细介绍在安装过程中的操作步骤。
经过上述流程图的介绍我们心中对安装过程有了一个大致的轮廓。下面我将对每一步进行细致的说明
步骤Ⅰ:检查显卡支持的的CUDA版本
我们可以通过英特尔的显卡控制面板来查看我们显卡驱动支持的CUDA版本
首先鼠标右击桌面-显示更多选项-NVIDIA控制面板-点击弹出界面左下角的(系统信息)-点击弹出界面的(组件)
经过上述操作后我们可以观测到红色方框内的信息可以看到(NVIDIA CUDA 12.0.89 driver)就是我电脑支持CUDA的版本,当然你电脑支持的版本可能和我不一样如果版本过低可以进行更新显卡驱动到这里我们就完成了步骤Ⅰ
步骤Ⅱ:下载对应的版本的CUDA