深度学习之CUDA/cudnn安装

导语:在安装好显卡驱动之后,就CUDA和cudnn的安装了。CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。(百度百科全文),cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。下面我们就进入CUDA的安装过程。


目录

CUDA安装

cudnn安装


NVIDIA

tf安装

显卡驱动安装


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__)

 运行结果如图即安装成功

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值