共需要安装以下几个部分:
1、nvidia驱动
2、CUDA
3、cuDNN
4、如果使用tensorflow,则安装对应的tensorflow-gpu版本
一、安装驱动
驱动这就不说了,直接安装最新的版本
二、安装CUDA
首先检查电脑支持的GPU版本
cmd命令行输入:nvidia-smi
这里cuda最高支持11.6,因此安装的时候要小于这个版本
CUDA的版本决定了CUDNN以及后面要安装的tensorflow-gpu的版本
第一步:下载CUDA
安装步骤:
检测是否安装好:
运行CMD程序,输入nvcc -V,显示如下,表示安装成功:
三、下载安装CUDNN
提示:可以复制链接到迅雷下载,无需注册账号
检验CUDNN是否安装成功:
版权声明:本文为m0_46423830原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。