--------------------------------------------GPU 相关安装-----------------------------------------------
Step1 更新最新的nvdia显卡驱动
功能:硬件的驱动
查看驱动支持的cuda版本
Step2 安装对应的cuda toolkit
功能:cuda的运算平台
官网地址:CUDA Toolkit Archive | NVIDIA Developer
注意:此处版本小于等于驱动支持的版本(版本向前兼容,即小于等于该版本的都行)
安装时记录下安装位置,下面会用到。安装时选择加到 环境变量 path中。
Step3 下载对应版本toolkit的cuDnn
功能:支持深度学习的运算包
官网地址cuDNN Archive | NVIDIA Developer
注意此处要适配对应的cuda toolkit。解压后覆盖到 cuda toolkit的路径下
附录:显卡算力查询 官网地址:CUDA GPUs - Compute Capability | NVIDIA Developer
至此GPU相关模块安装完毕。
本章节使用所用深度学习平台。因此独立说明。