通用安装 tensorflow GPU版环境方法

最新推荐文章于 2023-05-27 16:10:23 发布

深度探寻者

最新推荐文章于 2023-05-27 16:10:23 发布

阅读量469

点赞数

分类专栏： tensorflow教程文章标签： tensorflow 人工智能 python

本文链接：https://blog.csdn.net/m0_43505377/article/details/103671286

版权

tensorflow教程专栏收录该内容

3 篇文章 0 订阅

订阅专栏

TensorFlow有CPU版本和GPU版本之分，CPU版本安装相对简单，按着TensorFlow的官方文档进行安装即可。但CPU版本只能使用CPU进行计算，计算效率低。对于简单的模型计算可以使用CPU模式，但对于复杂的模型训练就需要GPU的支持了。

一、基本环境要求：

系统：windows 、Mac 、Linux 都可
显卡：一般为 NVIDIA 显卡

二、安装环境：

1、安装Anconda：
Anaconda是由Python提供支持的领先的开放数据科学平台。 Anaconda的开源版本是Python和R的高性能分发版本，包括超过100个用于数据科学的最流行的Python，R和Scala软件包。并且支持一键安装各种GPU加速包，比如 tensorflow gpu版。
选择自己对应的系统，建议Python3.7版本
官方网站下载：点击打开
在这里插入图片描述
2、CUDA环境
CUDA（Compute Unified Device Architecture），是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。它包含了CUDA指令集架构（ISA）以及GPU内部的并行计算引擎。
CUDA 是 Tensorflow GPU 的基础
首先查看自己电脑的显卡适合什么版本的CUDA（一般为CUDA10）
在这里插入图片描述
Windows系统查看
 官方