当前位置:网站首页>深度学习——安装CUDA以及CUDNN实现tensorflow的GPU运行
深度学习——安装CUDA以及CUDNN实现tensorflow的GPU运行
2022-08-03 15:11:00 【尘心平】
活动地址:CSDN21天学习挑战赛
这里我使用的版本是 tensorflow 2.8.0 tensorflow-gpu 2.8.0
目录
下载CUDA
查看NVIDIA 显卡支持的CUDA版本
- win+s 搜索 NVIDIA Console Panel并打开
2. 查看NVIDIA控制面板的系统信息(系统信息在下图左下角处)
打开系统信息窗口后,首先在显示部分查看并记录驱动程序版本
我的版本是512.15
然后点击组件,查看支持的CUDA版本
这里我的版本如下为11.6.127,下载11.6.1的CUDA
官网下载CUDA
- Bing或Google 搜索 CUDA 11.6.1 (这里你要下载哪个版本就输入 CUDA 版本号)
2. 点击链接,进入官网下载页面(可能网页加载会比较慢 VPN可能会快)
网页加载完后,页面如下:
这里根据目前所用的操作系统,选择Windows平台等信息
信息输入完成后,点击Download开始下载
这里我的浏览器设置为每次下载都要确定下载保存位置,所以会弹出下列窗口,点击保存即可
安装CUDA
创建下载目录文件夹
首先在软件保存的路径(自定义)处,创建CUDA和CUDA_program文件夹
运行下载的CUDA安装程序
- 点击之前下载好的cuda exe程序
这里提取文件路径选择之前创建的CUDA文件夹,之后OK即可
进度条加载完后,则会进入正式的CUDA 安装程序
- 正式的CUDA 安装程序
点击同意并继续后,进入许可协议页面
选择自定义安装,点击下一步,进入安装的软件选项
具体选择情况如下:
点击下一步,进入安装路径选择页面,这里的安装路径选择之前创建的CUDA_program路径
浏览修改好路径后,点击下一步,开始安装
根据自己的选择,之后Next即可(这里我是选择了 I understand 然后Next)
开始安装
安装结束,下一步即可
下载CUDNN
确定CUDNN版本
目前官网上并没有针对于11.6的版本,网络检索得到11.6的CUDA可以配置8.3.2的CUDNN
但是下载后,出现了一些dll找不到,之后下载了8.3.3的CUDNN发现可以解决之前的问题
因此 11.6.1的CUDA配置 8.3.3的CUDNN
官网下载CUDNN
CUDA和CUDNN的下载结果不同,CUDA是一个安装程序,而CUDNN则是个压缩文件
进入官网后,会需要登录NVIDIA账号,这里没有的去注册即可
登录后,可以看见各种CUDNN版本
这里我们下载8.3.3 For CUDA 11.6的版本
复制CUDNN解压文件
解压下载后的压缩包文件,解压后的文件夹包含下列文件
将解压后的文件夹复制到CUDA安装目录的bin文件夹的同级目录下并重命名为cudnn
这里我的是E:\SoftWares\PyCharm\CUDA_program路径下
添加环境变量
win+s输入环境变量,打开编辑系统环境变量
在系统环境变量Path中添加以下内容(具体是红框内的文件夹,按你自己的安装目录配置)
下载缺失的zlibwapi.dll文件
打开官网,下载zlibwapi.dll文件
一般的浏览器,点击ZLIB DLL 则会自动跳转下载;Chrome需要鼠标右键点击链接,选择链接另存为即可下载
下载完成后解压,并将解压后的文件夹zlib123dllx64移动到系统盘C的Program Files (x86)文件夹中
在系统变量Path中添加环境变量:
判断是否安装成功
- cmd命令行输入 nvcc -V 命令,出现下列信息则成功
- Pycharm运行下列脚本
from tensorflow.python.client import device_lib import tensorflow as tf print(device_lib.list_local_devices())
输出中的device_type既有CPU,也有GPU则成功
- 运行某神经网络程序
没有报错信息,能够正常epoch、正常运行即成功
边栏推荐
- 使用Typora+EasyBlogImageForTypora写博客,无图床快速上传图片
- php类的析构函数:__destruct
- 2021年12月电子学会图形化四级编程题解析含答案:质数判断器
- 2022年镇海夏令营组合数学和数论班 —— 数学作业 1
- 不安装运行时运行.NET程序
- 【指针内功修炼】函数指针 + 函数指针数组 + 回调函数(二)
- 力扣1206. 设计跳表--SkipList跳表是怎么跳的?
- HDU 1027 Ignatius and the Princess II(求由1-n组成按字典序排序的第m个序列)
- 又有大厂员工连续加班倒下/ 百度搜狗取消快照/ 马斯克生父不为他骄傲...今日更多新鲜事在此...
- 随笔-Unity中一个简易的Spine动画控制器
猜你喜欢
Mysql 生成排序序号
2021年12月电子学会图形化三级编程题解析含答案:跳高比赛
苹果开发「AI 建筑师」GAUDI:根据文本生成超逼真 3D 场景!
Currency ATM: Solana Wallet Has Unknown Security Vulnerability, A Large Number Of Users' Digital Assets Are Stolen
Day2:面试必考题目
王守创:多组学整合分析揭示植物代谢多样性的分子机制(8月2号晚)
技术分享 | 接口自动化测试如何搞定 json 响应断言?
问题6:下拉框测试点
云硬盘EVS详解以及如何用与避坑【华为云至简致远】
今日睡眠质量记录75分
随机推荐
【R语言科研绘图】--- 柱状图
跨桌面端之组件化实践
LeetCode136:只出现一次的数字
Mysql 生成排序序号
Detailed explanation of cloud hard disk EVS and how to use and avoid pits [HUAWEI CLOUD is simple and far]
问题6:下拉框测试点
问题5:发现缺陷怎么办?缺陷的类型有哪些?
R7 6800H+RTX3050+120Hz 2.8K OLED screen, Intrepid Pro15 2022 pre-sale
问题7:功能测试花瓶用例
文件包含之伪协议的使用
PostgreSQL V14中更好的SQL函数
ubiquant量化竞赛
问题8:对朋友圈进行用例设计
【软件工程之美 - 专栏笔记】36 | DevOps工程师到底要做什么事情?
Several methods of installing Mysql in Linux
问题9:为什么有边界值的测试?
问题4:什么是缺陷?你们公司缺陷的优先级是怎样划分的?
HDU 1406 (完数)
gocron定时任务管理系统的安装与运行
自己悦表存心