tensorflow cuda gpu 安装

news2024/11/25 15:25:46

Windows 安装 CUDA/cuDNN

Windows 安装 CUDA/cuDNN

需要注意的是一定要选择 TensorFlow 和 CUDA相匹配,还需要查看下自己GPU的驱动版本,如果不匹配会出现很多问题。GPU驱动的版本可在 NVIDIA控制面板里找到:

CUDA个版本与驱动的关系如下:

GPU版本的 TensorFlow 与CUDA和cuDNN的对应版本关系可以参考:https://www.tensorflow.org/install/source#tested_build_configurations。

可以看到如果选择 CUDA10.0 那么对应的 cuDNN是7.4,TensorFlow则需要1.13-2.0版本。

下载 CUDA

去 NVIDIA 下载 CUDA,下载地址为:https://developer.nvidia.com/cuda-downloads。本次使用的是 Windows10 64位,因此此处选择 Windows10 64位版本,为了安装方便选择 local 版本。默认的下载是最新版本的 CUDA,此时最新的是10.2,此处选择下载最新版本

若要下载其他版本的 CUDA,可以点击下面的 Legacy Releases,下载自己需要的 CUDA:

安装CUDA

下载安装软件之后,双击安装,会弹出选择提取安装保存位置,默认即可

之后开始提取安装文件

提取完成之后弹出安装界面

检查系统兼容性需要一点时间,之后按照提示一步步安装

接下来有两种安装选项,一个是精简安装,一个是自定义安装。精简安装会安装CUDA相关组件,同时也会将显卡驱动重新安装,如果不想重新安装显卡驱动,可以选择自定义安装,这里选择自定义安装

之后选择需要安装的组件,这里将驱动组件取消,其他保持勾选

之后会让选择CUDA开发组件、文档、示例的安装位置,此处默认即可,默认安装位置在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\版本

同意相关协议之后,进行安装前的准备,之后安装,安装需要一些时间:

如果电脑安装了杀毒软件,安装的过程中会出现如下提示,允许所有程序即可。

安装完成,提示如下:

根据需要,下面的可勾选也可以不勾选

安装完成CUDA,使用 nvcc -V 验证是否安装成功,看到如下信息说明安装成功

接下来就可以安装 cuDNN 了。

安装cuDNN

下载 cuDNN,下载之前需要先注册一下 Nvidia 的账号,下载地址为:https://developer.nvidia.com/rdp/cudnn-download 根据上面的对应关系,此处下载7.4版本:

选择Windows10版本下载

下载完成之后将其解压,解压之后的目录如下:

需要将以上三个文件复制到CUDA的安装目录中,通过上面的安装,我们将CUDA安装到了 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2目录中:

那么我们需要将:

  • cuDNN中bin目录下的文件移动到 CUDA 的 bin 目录中

  • cuDNN目录中的 include 中的文件移动到 CUDA 的 include 目录中
  • cuDNN目录中的 lib 中的文件移动到 CUDA 的 lib 目录中

这样就完成了cuDNN的安装。

验证安装

通过NVIDIA提供的 deviceQuery.exe 和 bandwidthTest.exe 来查看GPU的状态,两者均在安装目录的 extras\demo_suite文件夹中

首先运行 deviceQuery.exe ,可以看到如下结果:

运行 bandwidthTest.exe 看到如下结果:

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1009417.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

陷波器设计

中心频率 f c H z f_c \rm Hz fc​Hz 3dB陷波器带宽 f b H z f_b \rm Hz fb​Hz 陷波器深度 d d d 倍,即 20 log ⁡ ( d ) d B 20\log(d) \rm dB 20log(d)dB 则 B 2 π f b ω n 2 π f c \begin{aligned} B &2\pi f_b \\ \omega_n & 2\pi f_c \end…

「中秋来袭」没想到,用OpenCV竟能画出这么漂亮的月饼「附源码」

一、前言 中秋佳节即将来临,作为传统的中国节日之一,人们除了品尝美味的月饼、赏月外,还喜欢通过绘画来表达对这个节日的喜悦和祝福。而如今,随着科技的不断发展,竟然可以借助计算机视觉库OpenCV来绘制精美的月饼和可…

Flutter的oktoast插件详解

文章目录 简介详细介绍安装和导入导入在MaterialApp外面套一层OKToast组件为什么是包住MaterialApp? 显示Toast消息: 高级使用Toast位置Toast持续时间自定义Toast样式高级用法 使用场景提示消息表单验证操作反馈网络请求状态调试信息小结 总结 简介 okt…

剑指offer刷题笔记 开端

剑指offer刷题笔记 文章目录 剑指offer刷题笔记注意内容 时间复杂度C 语法知识补充:优先级new 和 delete树的遍历 算法模板string类判断字符串为空归并递归整数二分算法模板 注意内容 map 是 O(logn) ,底层实现是平衡树,unorder_map 是 O(1)…

Thinkphp6 配置并使用redis图文详解 小皮面板

这篇文章主要介绍了Thinkphp6 配置并使用redis的方法,结合实例形式详细分析了Redis的安装、配置以及thinkphp6操作Redis的基本技巧,需要的朋友可以参考下 一、安装redis ThinkPHP内置支持的缓存类型包括file、memcache、wincache、sqlite。ThinkPHP默认使用自带的采用think\Ca…

zabbix监控多实例redis

Zabbix监控多实例Redis 软件名称软件版本Zabbix Server6.0.17Zabbix Agent5.4.1Redis6.2.10 Zabbix客户端配置 编辑自动发现脚本 vim /usr/local/zabbix/scripts/redis_discovery.sh #!/bin/bash #Fucation:redis low-level discovery #Script_name redis_discovery.sh red…

软件外包开发原型图工具

国内有一些原型图管理工具,这些工具旨在帮助团队创建、协作和管理应用程序的原型图。以下是一些国内的原型图管理工具及其特点,希望对大家有所帮助。北京木奇移动技术有限公司,专业的软件外包开发公司,欢迎交流合作。 1.墨刀&…

学习视觉SLAM需要会些什么?

前言 SLAM是现阶段很多研究生的研究方向,我也是作为一个即将步入视觉SLAM的研究生,网上对于SLAM的介绍很多,但很少有人完整系统的告诉你学习视觉SLAM该有那些基础,那么此贴将告诉你学习SLAM你要有那些方面的基础。 文章目录 前言…

在Linux上利用宝塔面板搭建网站,并通过内网穿透方便地实现公网访问

文章目录 前言1. 环境安装2. 安装cpolar内网穿透3. 内网穿透4. 固定http地址5. 配置二级子域名6. 创建一个测试页面 前言 宝塔面板作为简单好用的服务器运维管理面板,它支持Linux/Windows系统,我们可用它来一键配置LAMP/LNMP环境、网站、数据库、FTP等&…

大数据课程M2——ELK的ELASTICSEARCH概述

文章作者邮箱:yugongshiye@sina.cn 地址:广东惠州 ▲ 本章节目的 ⚪ 了解ELK的ELASTICSEARCH概括; ⚪ 掌握ELK的ELASTICSEARCH核心概念; 一、ELASTICSEARCH概括 1. ES安装和启动 注意:如果使用达内云主机,无需安装es,直接配置yml启动即可。 1. 安装 下…

普中51-蜂鸣器实验

蜂鸣器实验 蜂鸣器主要分为压电式蜂鸣器和电磁式蜂鸣器两种类型 压电式蜂鸣器主要由多谐振荡器、压电蜂鸣片、阻抗匹配器及共鸣箱、外壳等组成。多谐振荡器由晶体管或集成电路构成,当接通电源后(1.5~15V 直流工 作电压),多谐振荡…

量化INT8模型报错记录及解决方案

基于C将onnx 模型 转 TensorRT 模型(int8) 部分命令如下: config->setFlag(nvinfer1::BuilderFlag::kINT8); ICudaEngine* engine builder->buildEngineWithConfig(*network, *config); 出现以下报错: Error Code 4: I…

使用凌鲨进行聚合搜索

作为研发人员,我们经常需要在多个来源之间查找信息,以便进行研发工作。除了常用的搜索引擎如百度和必应之外,我们还需要查阅各种代码文档和依赖包等资源。这些资源通常分散在各个网站和文档库中,需要花费一定的时间和精力才能找到…

闲置的固定资产怎么账务管理

对于那些被遗忘或闲置的固定资产,如何进行有效的账务管理,以实现资产的最大价值,却往往被忽视。本文旨在探讨这一主题,并提供一些创新的策略和实践方法。  需要认识到,闲置的固定资产并非仅仅是财务问题。它们可能代…

【css | loading】各种简单的loading特效

先看一下简单的效果图 线上预览 https://code.juejin.cn/pen/7278511766991339579 代码 HTML <!-- / 1 --><section><div class"sk-rotating-plane"></div></section><!-- / 2 --><section><div class"sk-doub…

OOM分析实战

OOM分析&实战 OOM分析&实战引言&#xff1a;一、JVM内存结构二、JVM OOM错误情况三、实践案例一案例二案例三 四、总结五、分析工具推荐六、参考文献 OOM分析&实战 引言&#xff1a; 在Java开发中&#xff0c;随着应用程序变得越来越复杂&#xff0c;内存管理问题…

运动耳机哪个好、最好的运动牌子排名榜

很多朋友喜欢在运动的时候听音乐&#xff0c;为此&#xff0c;他们会为自己配备一款蓝牙耳机或是运动耳机&#xff0c;可以在运动的时候随身听&#xff0c;可是&#xff0c;一些人在挑选耳机的时候犯难了&#xff0c;市面上那么多运动耳机&#xff0c;运动耳机哪个好&#xff1…

【黄色手套22】6话:构造数据类型

目录 序言&#xff1a; 结构体&#xff1a; 结构体数组&#xff1a; 结构体数组的定义和引用 &#xff1a; 使用关键字typedef自定义类型名&#xff1a; 使用typedef语句为数组类型取别名&#xff1a; 结构体所占内存的大小&#xff1a; 结构体的嵌套&#xff1a; 共用体…

内网穿透——Windows搭建服务器

文章目录 1.前言2. Emby网站搭建2.1. Emby下载和安装2.2 Emby网页测试 3. 本地网页发布3.1 注册并安装cpolar内网穿透3.2 Cpolar云端设置3.3 Cpolar内网穿透本地设置 4.公网访问测试5.结语 1.前言 在现代五花八门的网络应用场景中&#xff0c;观看视频绝对是主力应用场景之一&…

vant 按需导入 vue2

vant 按需导入 vue2 1、通过npm安装 # Vue 3 项目&#xff0c;安装最新版 Vant&#xff1a; npm i vant -S# Vue 2 项目&#xff0c;安装 Vant 2&#xff1a; npm i vantlatest-v2 -S2、自动按需引入组件 babel-plugin-import 是一款 babel 插件&#xff0c;它会在编译过程中…