确认是否有可供CUDA使用的GPU
可通过【windows设备管理器】中的【显示适配器】中查看。
【设备管理器】可通过在命令行窗口输入以下命令打开:
control /name Microsoft.DeviceManager
显示适配器中展示了显卡的型号,如下图:
显卡型号上右键,选择【属性】,然后再【属性】窗口中切换到【驱动程序】即可查看驱动版本。
下载CUDA安装包
这里有不同版本CUDA的安装包及文档。
可以选择下载不同类型的安装包:
Network Installer:一个最小的安装程序,大概只有十几兆,安装过程中再下载选择的需要的部分软件包。 (推荐方式,省时间)。
Local Installer:包含 CUDA 工具包的所有组件,有2个多G大小。
安装CUDA
检查CUDA是否可用
继续下一步之前,需要验证 CUDA 工具包是否可以找到支持 CUDA 的硬件并与之正确通信。
更多详细内容参考NVIDIA CUDA安装文档。