导语:在安装好显卡驱动之后,就CUDA和cudnn的安装了。CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。(百度百科全文),cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。下面我们就进入CUDA的安装过程。
目录
CUDA安装
进入NVIDIA官网>驱动程序>开发者>CUDA>download now
注:之前在安装tensorflow时,安装了2.60.版本 的,对应CUDA的11.2版本,不要直接下载
点击下载,等待下载完成
进入安装界面>系统检查兼容>同意许可>选择自定义安装>保留两个选项>更改文件位置
安装完成退出安装程序
cudnn安装
进入CUDA下载页面>Platform>cudnn>download cuDNN>Archived cuDNN
注:安装tf2时,适配cuDNN版本是8.1
下载完成后,在与CUDA同一块盘中创建cuDNN8.1文件夹,将下载好的压缩包解压到文件夹
将cuDNN的bin文件夹和include文件夹中的内容分别复制到CUDA的bin和include文件夹下
全部完成后打开pycharm
import tensorflow as tf
print(tf.__version__)
运行结果如图即安装成功