怎么安装tensorflow？

时间：2023-10-26 17:18:01 | 来源：网站运营

时间：2023-10-26 17:18:01 来源：网站运营

怎么安装tensorflow？：

前情提要：

这几天实验室新配了一台3090的工作站，说是工作站，其实是游戏主机（GPU实在是太难买了！）。所以跟随主机过来的是Windows系统。考虑到保修协议，一年内不能刷成Linux系统，就只能琢磨一下怎么在Windows配置TensorFlow了。此文用于记录在我尝试配置成功后的每一步，也分享给有需要的大家。

为什么需要一个教程？

因为如果你按照官方pip一键安装：

pip install tensorflow你一定会发现你的模型没有在使用GPU。其实官方也说得很清楚，需要你Build from source。

为什么不用Conda命令来安装?

Conda很好，按照这个命令，确实可以安装出能使用GPU的版本

conda install -c anaconda tensorflow-gpu因为Conda目前只更新到2.6（我自己测试了一下，给我装了2.1，我很迷惑），所以如果你对于版本要求没有很严格，可以采取这个方案，请点赞然后不用阅读后面的啦。

本文适合谁？

需要TensorFlow-GPU 2.6+的使用Windows系统的同学。需要一定的基础知识，比如命令行的使用。

声明：

本文主要根据这两个信息源：

没有墙且英语好的同学也可以直接去这两个教程。

开始吧。

第一步：更新你的NVIDIA驱动

这一步因为我是GeForce RTX系列，所以我直接使用NVIDIA GeForce Experience软件完成安装。

使用别的系列的卡的同学可以去 Download the latest official NVIDIA drivers 下载驱动然后安装。

第二步：安装Visual Studio

这一步骤的意义根据Jeff Heaton教授所说是为了让我们的系统有一个可用的C++编译器。注意是Visual Studio，不是VS Code。

在安装的时候不用选任何的Workload，Jeff建议选择如下Individual Components:

点击Install然后会有一个这个框框：

因为我们不需要Workload，所以直接点击Continue就可以开始安装了。

16MB每秒的速度，还可以。

安装完成后找到安装目录，我的在这里：C:/Program Files/Microsoft Visual Studio

找到这个文件：C:/Program Files/Microsoft Visual Studio/2022/Community/VC/Auxiliary/Build/vcvars64

打开CMD，输入：

cd C:/Program Files/Microsoft Visual Studio/2022/Community/VC/Auxiliary/Build/然后输入：

vcvars64.bat你应当看到如下信息：

说明咱们的C++编译器就设定好了。

第三步：安装CUDA11.2

虽然已经有11.6，但是目前官方还是推荐使用CUDA11.2+CuDNN8.1的配置。

去这里找11.2的安装包：

CUDA Toolkit Archive下载然后安装。

我装的是11.2.2，这个安装包很无脑，选择Express安装然后按照默认一路点下去就好了。

第四步：安装CuDNN8.1

这一步就很麻烦了，首先你要在这里下载CuDNN 8.1.1的 .zip文件：

NVIDIA会让你注册一个开发者账号然后才能下载。（据我所知墙内的同学可以使用XX镜像？）

点击我下划线的那个，你会下载一个压缩文件。重点来了，下面是我配置成功的技巧，可能还可以化简，但是我就这么配置了：

1. 你需要在你的C盘建立一个叫做tools的文件夹。
2. 解压刚刚下载的压缩文件到tools里面。
3. 打开系统的路径管理器，长这样：

4. 打开Environment Variables，在System Variables找到Path，然后点击Edit。

5.将以下内容加入进你的Path中：

C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2/binC:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2/extras/CUPTI/lib64C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2/extras/CUPTI/includeC:/tools/cudnn-11.2-windows-x64-v8.1.1.33/cuda/binC:/tools/cudnn-11.2-windows-x64-v8.1.1.33/cuda/include注意检查一下，大部分同学应该在path里有第一条了。同时这里的路径是我自己的（一般也是默认安装后的），如果不一致需要自己更改成你们的。

第五步：确认PATH

这一步很重要，打开你们的CMD，按照这个方式挨个测试你们的系统能访问上面五个路径：

dir "PATH"把“”里面的PATH换成每个path，比如这样：

如果有这样的信息，说明路径是可以访问的，没有问题。

第六步：复制dll文件

有些人说这个没必要，但是我建议还是这么做，算是双保险。

1. 把C:/tools/cudnn-11.2-windows-x64-v8.1.1.33/cuda/bin里的全部.dll文件复制
2. 粘贴到C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.2/bin

第七步：安装TensorFlow 2.8

终于到这一步了。

建议用Conda建立一个新的环境，我的叫做tf2:

conda create --name tf2conda activate tf2然后

pip install tensorflow等待安装完成后，测试一下：

成功！

结尾的一些话

这是我第一次在本站写技术类长文，还有很多不妥、讲的不细节的地方，希望大家多多包涵！