Win10安装配置Tensorflow2 + NVIDIA GPU
1. Python环境安装
首先从Anaconda官网(https://www.anaconda.com/)下载Anaconda 安装文件。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第1张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第1张](https://www.buguit.com/wp-content/uploads/2023/09/image-45-1024x460.png)
下载对应系统的安装文件,运行安装。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第2张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第2张](https://www.buguit.com/wp-content/uploads/2023/09/image-46-1024x648.png)
安装完成后,打开终端,进入Anaconda虚拟环境,默认的base环境是python3。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第3张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第3张](https://www.buguit.com/wp-content/uploads/2023/09/image-48-1024x392.png)
2. Tensorflow安装
(1)新建Anaconda虚拟环境
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第4张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第4张](https://www.buguit.com/wp-content/uploads/2023/09/image-49-1024x565.png)
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第5张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第5张](https://www.buguit.com/wp-content/uploads/2023/09/image-50-1024x580.png)
Python版本不要太高,这里选择3.7。
(2)安装Tensorflow-gpu
需要注意:
a. 从2.0版本开始,CPU和GPU软件包合在一起的(1.x的版本是分开的);
b. 从2.11版本开始,TensorFlow不再支持Windows上的GPU版本,需要在windows WLS2上安装才能使用GPU版本。 因此这里我们需要安装2.0~2.10之间的版本,这里我们选择2.10.0版本。
安装方式一:Anaconda安装
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第6张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第6张](https://www.buguit.com/wp-content/uploads/2023/09/image-51-1024x496.png)
安装方式二:pip安装
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第7张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第7张](https://www.buguit.com/wp-content/uploads/2023/09/image-52-1024x402.png)
终端查看安装的版本(由于没有安装CUDA,这里会提示dlerror。):
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第8张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第8张](https://www.buguit.com/wp-content/uploads/2023/09/image-53-1024x336.png)
3. 安装NVIDIA显卡驱动
到NVIDIA官网(https://www.nvidia.cn/Download/index.aspx?lang=cn)下载对应显卡驱动。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第9张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第9张](https://www.buguit.com/wp-content/uploads/2023/09/image-54-1024x656.png)
安装完成后,可以在命令行执行“nvidia-smi”命令查看显卡信息。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第10张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第10张](https://www.buguit.com/wp-content/uploads/2023/09/image-55-1024x737.png)
4. CUDA安装
CUDA(Compute Unified Device Architecture)是NVIDIA的通用并行计算架构,需要安装CUDA后,才能使用TensorFlow的GPU加速计算能力。在安装CUDA之前注意其与TensorFlow的版本兼容关系(https://tensorflow.google.cn/install/source_windows?hl=en#gpu)。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第11张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第11张](https://www.buguit.com/wp-content/uploads/2023/09/image-56-1024x777.png)
对应TensorFlow2.10的CUDA版本是11.2。到官网(https://developer.nvidia.com/cuda-toolkit-archive)下载。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第12张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第12张](https://www.buguit.com/wp-content/uploads/2023/09/image-57-1024x992.png)
下载完成后,直接双击安装文件,默认配置完成安装。安装完成后,终端输入“nvcc -V”,查看CUDA版本。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第13张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第13张](https://www.buguit.com/wp-content/uploads/2023/09/image-58-1024x233.png)
5. cuDNN安装
官网(https://developer.nvidia.com/rdp/cudnn-archive)下载,注意下载和CUDA对应的版本。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第14张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第14张](https://www.buguit.com/wp-content/uploads/2023/09/image-59-1024x477.png)
下载之后,解压压缩包,得到下图所示3个文件夹:
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第15张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第15张](https://www.buguit.com/wp-content/uploads/2023/09/image-60-1024x279.png)
将以上3个文件夹复制到CUDA的安装目录即可。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第16张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第16张](https://www.buguit.com/wp-content/uploads/2023/09/image-61-1024x440.png)
6. Demo
在pycharm中新建项目。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第17张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第17张](https://www.buguit.com/wp-content/uploads/2023/09/image-62-1024x848.png)
选择Anaconda中创建的环境。
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第18张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第18张](https://www.buguit.com/wp-content/uploads/2023/09/image-63-1024x677.png)
测试代码:
import tensorflow as tf
print(tf.config.list_physical_devices())
print(tf.config.list_physical_devices('GPU'))
print(tf.config.list_physical_devices('CPU'))
Code language: PHP (php)
运行结果如下:
![Win10安装配置Tensorflow2 + NVIDIA GPU - 第19张 - 布谷IT Win10安装配置Tensorflow2 + NVIDIA GPU - 第19张](https://www.buguit.com/wp-content/uploads/2023/09/image-64-1024x342.png)
完成!