Tensorflow安装与配置教程(超详细,细的有点烦)

2024-02-18 15:20

本文主要是介绍Tensorflow安装与配置教程(超详细,细的有点烦),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

一、在安装tensorflow-gpu2.1.0之前,首先需要确定我们的电脑到底支不支持gpu,具体的查看方式如下图所示:

首先选中“我的电脑”单击右键---->“管理”---->“设备管理器”---->“显示适配器”---->查看是否有NVIDIA显卡,要是有就表示我们可以安装tensorflow-gpu,否则我们只能安装不支持gpu的tensorflow版本。

 二、估计你需要安装的tensorflow-gpu版本,然后查看tensorflow与CUDA的版本对应关系(本教程安装tensorflow-gpu==2.1.0)

根据上表查找对应的cuda版本号,这里选择cuda10.1。当确定cuda版本号以后进行cuda的下载。

三、下载cuda

建议大家最好在cuda官网去下载(CUDA Toolkit Archive | NVIDIA Developer 

 选择cuda10.1版本进行下载

 大家可以根据自己的操作系统勾选对应的选项进行下载。

四、安装cuda

在win10系统下cuda的安装与普通软件的安装大体一致,只是在安装的过程中有一些地方需要注意。

 *这里大家一定要注意,尽量选择自定义安装,因为后续还有版本需要核对。

 进入这一步后,依次点开CUDA、Other components、Driver components前面的加号,然后核对每一项的“新版本”与“当前版本”值。当“新版本”的值小于“当前版本”时去掉前面的勾选,否则会造成安装不成功。如果“新版本”的值大于“当前版本”可以勾选前面的选项。(非常重要)

当选择完这一步后,接下来持续点击,直到安装完成。

五、环境变量配置

至此我们已经将cuda安装完成了,现在需要进行环境变量的配置并验证安装是否有效。

依次在cuda的安装目录中配置如上图所示红框的环境变量。配置完成后打开终端并输入nvcc -V(注意:是大写V,不要输入小写)来验证是否安装成功。

当看到如上信息,表示cuda已经安装成功了。 

 六、cudnn的安装

NVIDIA cudnn是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。cudnn是基于cuda的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。cudnn的安装相对比较简单。首先去下载cudnn,可以去官方网站(cuDNN Archive | NVIDIA Developer下载,麻烦就在于需要创建账户。大家也可以在清华镜像(https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/)中去下载cudnn。

 下载完成后进行解压,解压完成后会看到下图所示中的文件。

 然后进入Library文件夹,可以看到如下图所示的三个文件夹。

 将Library文件夹下的bin、include、lib三个文件夹复制到cuda的安装目录下并替换原有的文件夹。

 至此,我们就完成了cuda、cdnn的安装。

七、tensorflow-gpu2.3.0的安装

当完成了前面cuda以及cudnn的安装后,就可以进行tensorflow-gpu的安装了。打开命令行,然后输入pip install tensorflow-gpu==2.3.0进行安装即可。如在安装的过程中遇到问题,请自行寻找解决方案。

 

 安装完成后,测试安装的tensorflow能不能调用gpu

***   如果出现“ DLL load failed: 找不到指定的模块”的错误,可以在网站https://docs.microsoft.com/zh-CN/cpp/windows/latest-supported-vc-redist?view=msvc-170

下载对应的版本进行安装。具体可参考博客:

https://blog.csdn.net/sereasuesue/article/details/105327611

八、pytorch的安装及测试

在安装pytorch之前,大家首先需要确认pytorch与cuda的版本对应关系,可以参考网址

Previous PyTorch Versions | PyTorch来进行核对。这里我就直接选择pytorch1.9+cuda10.2进行安装。首先打开pytorch官网:

大家可以根据自己的电脑系统以及安装的cuda版本选择对应的选项。当勾选完成后,复制下方的命令进行安装。 

然后指令粘贴在命令行下进行安装,这里我将pip3修改为了pip。如果大家只装了anaconda的环境,这里可以不用修改。 

 安装的过程有点慢,大家也可以使用清华镜像来安装

-i https://pypi.tuna.tsinghua.edu.cn/simple

安装完成后即可进行测试,具体测试方法如下图所示:

至此我们就完成了全部的环境配置。如果大家在安装过程中遇到什么问题,可以留言也可以查找其他的解决方案。

八、参考博客

1、深度学习GPU环境CUDA详细安装过程(简单快速有效) - 知乎

2、深度学习GPU环境CUDA详细安装过程(简单快速有效) - 知乎

 非常感谢以上作者的优秀博文。

这篇关于Tensorflow安装与配置教程(超详细,细的有点烦)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/721584

相关文章

Flutter实现文字镂空效果的详细步骤

《Flutter实现文字镂空效果的详细步骤》:本文主要介绍如何使用Flutter实现文字镂空效果,包括创建基础应用结构、实现自定义绘制器、构建UI界面以及实现颜色选择按钮等步骤,并详细解析了混合模... 目录引言实现原理开始实现步骤1:创建基础应用结构步骤2:创建主屏幕步骤3:实现自定义绘制器步骤4:构建U

Android开发环境配置避坑指南

《Android开发环境配置避坑指南》本文主要介绍了Android开发环境配置过程中遇到的问题及解决方案,包括VPN注意事项、工具版本统一、Gerrit邮箱配置、Git拉取和提交代码、MergevsR... 目录网络环境:VPN 注意事项工具版本统一:android Studio & JDKGerrit的邮

Win安装MySQL8全过程

《Win安装MySQL8全过程》:本文主要介绍Win安装MySQL8全过程,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录Win安装mysql81、下载MySQL2、解压文件3、新建文件夹data,用于保存数据库数据文件4、在mysql根目录下新建文件my.ini

SpringBoot3.4配置校验新特性的用法详解

《SpringBoot3.4配置校验新特性的用法详解》SpringBoot3.4对配置校验支持进行了全面升级,这篇文章为大家详细介绍了一下它们的具体使用,文中的示例代码讲解详细,感兴趣的小伙伴可以参考... 目录基本用法示例定义配置类配置 application.yml注入使用嵌套对象与集合元素深度校验开发

springboot使用Scheduling实现动态增删启停定时任务教程

《springboot使用Scheduling实现动态增删启停定时任务教程》:本文主要介绍springboot使用Scheduling实现动态增删启停定时任务教程,具有很好的参考价值,希望对大家有... 目录1、配置定时任务需要的线程池2、创建ScheduledFuture的包装类3、注册定时任务,增加、删

IntelliJ IDEA 中配置 Spring MVC 环境的详细步骤及问题解决

《IntelliJIDEA中配置SpringMVC环境的详细步骤及问题解决》:本文主要介绍IntelliJIDEA中配置SpringMVC环境的详细步骤及问题解决,本文分步骤结合实例给大... 目录步骤 1:创建 Maven Web 项目步骤 2:添加 Spring MVC 依赖1、保存后执行2、将新的依赖

SpringBoot基于配置实现短信服务策略的动态切换

《SpringBoot基于配置实现短信服务策略的动态切换》这篇文章主要为大家详细介绍了SpringBoot在接入多个短信服务商(如阿里云、腾讯云、华为云)后,如何根据配置或环境切换使用不同的服务商,需... 目录目标功能示例配置(application.yml)配置类绑定短信发送策略接口示例:阿里云 & 腾

如何为Yarn配置国内源的详细教程

《如何为Yarn配置国内源的详细教程》在使用Yarn进行项目开发时,由于网络原因,直接使用官方源可能会导致下载速度慢或连接失败,配置国内源可以显著提高包的下载速度和稳定性,本文将详细介绍如何为Yarn... 目录一、查询当前使用的镜像源二、设置国内源1. 设置为淘宝镜像源2. 设置为其他国内源三、还原为官方

最详细安装 PostgreSQL方法及常见问题解决

《最详细安装PostgreSQL方法及常见问题解决》:本文主要介绍最详细安装PostgreSQL方法及常见问题解决,介绍了在Windows系统上安装PostgreSQL及Linux系统上安装Po... 目录一、在 Windows 系统上安装 PostgreSQL1. 下载 PostgreSQL 安装包2.

Maven如何手动安装依赖到本地仓库

《Maven如何手动安装依赖到本地仓库》:本文主要介绍Maven如何手动安装依赖到本地仓库问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、下载依赖二、安装 JAR 文件到本地仓库三、验证安装四、在项目中使用该依赖1、注意事项2、额外提示总结一、下载依赖登