在 Windows 系统上安装 TensorFlow-GPU 2.12.0
时间: 2024-06-09 22:12:03 浏览: 343
Windows安装GPU版TensorFlow
好的,以下是在 Windows 系统上安装 TensorFlow-GPU 2.12.0 的步骤:
1. 确认显卡驱动已经安装并支持 CUDA 10.0 或 10.1,这是 TensorFlow-GPU 2.12.0 的要求。
2. 安装 CUDA Toolkit 10.0 或 10.1,下载地址为:
* CUDA 10.0:https://developer.nvidia.com/cuda-10.0-download-archive
* CUDA 10.1:https://developer.nvidia.com/cuda-10.1-download-archive
安装时选择自定义选项,仅安装 CUDA 核心组件和 CUDA 工具。安装完成后需要设置环境变量:
```
CUDA_HOME = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 (或 v10.1)
Path = %PATH%;%CUDA_HOME%\bin;%CUDA_HOME%\extras\CUPTI\lib64
```
3. 安装 cuDNN 7.4.2 或 7.5.x,下载地址为:
* cuDNN 7.4.2:https://developer.nvidia.com/rdp/cudnn-archive
* cuDNN 7.5.x:https://developer.nvidia.com/rdp/cudnn-download
将文件解压到 CUDA Toolkit 的安装路径下,例如:
```
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
```
4. 安装 Anaconda,下载地址为:https://www.anaconda.com/products/individual
5. 创建一个新的 Python 环境:
```
conda create -n tf_gpu python=3.6
```
激活环境:
```
conda activate tf_gpu
```
6. 安装 TensorFlow-GPU 2.12.0:
```
pip install tensorflow-gpu==2.12.0
```
安装完成后可以测试是否成功,例如:
```
import tensorflow as tf
print(tf.__version__)
```
如果输出版本号为 2.12.0,说明安装成功。
希望这些步骤可以帮助你安装 TensorFlow-GPU 2.12.0。
阅读全文