GPU 相关安装

--------------------------------------------GPU 相关安装-----------------------------------------------

Step1 更新最新的nvdia显卡驱动

功能:硬件的驱动

驱动官网:官方 GeForce 驱动程序 | NVIDIA

GPU 相关安装_第1张图片

查看驱动支持的cuda版本

GPU 相关安装_第2张图片

Step2 安装对应的cuda toolkit

功能:cuda的运算平台

官网地址:CUDA Toolkit Archive | NVIDIA Developer

GPU 相关安装_第3张图片

注意:此处版本小于等于驱动支持的版本(版本向前兼容,即小于等于该版本的都行)

安装时记录下安装位置,下面会用到。安装时选择加到 环境变量 path中。

Step3 下载对应版本toolkitcuDnn

功能:支持深度学习的运算包

官网地址cuDNN Archive | NVIDIA Developer

GPU 相关安装_第4张图片

注意此处要适配对应的cuda toolkit。解压后覆盖到 cuda toolkit的路径下

附录:显卡算力查询 官网地址:CUDA GPUs - Compute Capability | NVIDIA Developer

GPU 相关安装_第5张图片

至此GPU相关模块安装完毕。

本章节使用所用深度学习平台。因此独立说明。

你可能感兴趣的:(pytorch,深度学习,人工智能,python,pytorch)