tensorflow安装GPU版本

简介: tensorflow安装GPU

tensorflow安装GPU版本主要要点

1.先通过该网站查看tensorflow和cuda和cudnn版本以及visual studio(MSVC)的对应关系。(可供参考)

https://www.tensorflow.org/install/source_windows#gpu

在英伟达控制面板,点击右下角的系统信息,可查看驱动的版本信息,点击上方的组件,可查看需要的cuda的版本(这里指的是你的电脑支持的最高版本的cuda driver版本,并不是推荐你安装这个,仅参考)

选择cuda的版本比较麻烦,需要兼顾tensorflow,显卡驱动版本,电脑版本。

可以参考一下这个,驱动版本和cuda的版本对应,只要驱动版本符合大于等于的要求就可以使用对应的cuda版本。

image.png

2.在这个网站上根据自己需要下载对应版本的cuda toolkit,如果网络很好可以选择下载较小的network包,一般建议安装local本地包。

https://developer.nvidia.com/cuda-toolkit-archive

安装时可以选择精简安装,也可以自定义安装,可选择想安装的插件,一般可以默认全选。

在安装完成后,会自动添加一部分cuda的环境变量

cuda的环境变量不需要再手动添加了。

注:可以看这篇文章了解cuda driver和cuda toolkit的区别,加深理解。

https://www.cnblogs.com/marsggbo/p/11838823.html

3.在这选择对应的cuda和tensorflow的版本的cudnn下载。

https://developer.nvidia.com/rdp/cudnn-archive

4.下载完成后,解压文件得到lib,bin,include和一个txt文件,将三个文件夹下的三个文件分别对应复制到cuda的安装路径下的对应文件夹下(默认安装的话是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0下的lib,bin,include文件夹下)

5.下面添加cudnn的环境变量,先新建一个系统变量,如下图,然后在系统变量的path下进行编辑,新建如第二张图的两个解析路径。

image.png

image.png

在安装后cuda后可以通过cmd输入nvcc -V验证一下。

在安装好tensorflow-gpu后,

执行下面代码,如果打印use GPU true,则代表gpu安装完成,可以使用gpu进行训练。

import tensorflow as tf

gpu_ok = tf.test.is_gpu_available()
print("tf version:", tf.__version__)
print("use GPU", gpu_ok)

我的环境以及最终安装完成的版本(rtx2060驱动版本451.67,cuda10.0.130,cudnn7.6.2.24,tensorflow-gpu1.15.4,win10x64)

仅供参考

相关实践学习
在云上部署ChatGLM2-6B大模型(GPU版)
ChatGLM2-6B是由智谱AI及清华KEG实验室于2023年6月发布的中英双语对话开源大模型。通过本实验,可以学习如何配置AIGC开发环境,如何部署ChatGLM2-6B大模型。
目录
相关文章
|
自然语言处理 语音技术
交大x-lance跨媒体语言智能实验室等联合开源F5-TTS!零样本语音复刻,生成流畅,情感丰富!
上海交大x-lance跨媒体语言智能实验室联合剑桥大学、吉利汽车研究院(宁波)公司开源了一种基于流匹配的扩散变换器(Diffusion Transformer,DiT)的完全非自回归TTS模型-F5-TTS。
|
TensorFlow 算法框架/工具 C++
在有GPU的windows上安装TensorFlow
在有GPU的windows上安装TensorFlow
777 0
|
9月前
|
机器学习/深度学习 人工智能 Cloud Native
Anaconda3 2025 深度解析:Python 3.11 + 提速、大模型支持,安装步骤+安装包
Anaconda3 2025 是面向数据科学与AI开发的一站式平台,集成Python环境、包管理与开发工具,支持机器学习、大模型开发及云原生部署,开箱即用,助力高效开发。
2467 2
|
存储 消息中间件 监控
Java日志详解:日志级别,优先级、配置文件、常见日志管理系统ELK、日志收集分析
Java日志详解:日志级别,优先级、配置文件、常见日志管理系统、日志收集分析。日志级别从小到大的关系(优先级从低到高): ALL < TRACE < DEBUG < INFO < WARN < ERROR < FATAL < OFF 低级别的会输出高级别的信息,高级别的不会输出低级别的信息
|
存储 算法 数据安全/隐私保护
手把手教你在FPGA上搭建一个ARM Cortex-M3软核
手把手教你在FPGA上搭建一个ARM Cortex-M3软核
2256 0
手把手教你在FPGA上搭建一个ARM Cortex-M3软核
|
机器学习/深度学习 弹性计算 Ubuntu
“炫技”还是“真硬核”,OpenPPL 实测阿里「倚天 710」芯片
本文将以「深度学习模型推理应用」为出发点,对「倚天 710」这款 ARM Server 芯片进行性能实测。
3030 0
“炫技”还是“真硬核”,OpenPPL 实测阿里「倚天 710」芯片
|
API Android开发
31. 【Android教程】状态栏通知:Notification
31. 【Android教程】状态栏通知:Notification
2118 1
|
大数据 vr&ar 芯片
世界杯的历史也是一部科技史
世界杯不只是足球圈儿的事。在与世界杯产生交集的众多角色里,一个不可忽视的角色是科技企业。
3347 0
|
机器学习/深度学习 计算机视觉
【小样本图像分割-1】PANet: Few-Shot Image Semantic Segmentation with Prototype Alignment
本文介绍了ICCV 2019的一篇关于小样本图像语义分割的论文《PANet: Few-Shot Image Semantic Segmentation With Prototype Alignment》。PANet通过度量学习方法,从支持集中的少量标注样本中学习类的原型表示,并通过非参数度量学习对查询图像进行分割。该方法在PASCAL-5i数据集上取得了显著的性能提升,1-shot和5-shot设置下的mIoU分别达到48.1%和55.7%。PANet还引入了原型对齐正则化,以提高模型的泛化能力。
991 0
【小样本图像分割-1】PANet: Few-Shot Image Semantic Segmentation with Prototype Alignment