tensorflow-gpu2.12.0安装
时间: 2023-09-27 09:10:15 浏览: 212
安装tensorflow-gpu
5星 · 资源好评率100%
以下是在 Windows 系统上安装 TensorFlow-GPU 2.1.0 的步骤:
1. 环境要求:
- NVIDIA® GPU card with CUDA® Compute Capability 3.5 or higher
- CUDA® Toolkit 10.1
- cuDNN SDK 7.6
- Python 3.6-3.8 (64-bit)
2. 安装 CUDA® Toolkit 10.1:
- 下载 CUDA® Toolkit 10.1 安装文件并运行,按照默认设置完成安装。
3. 安装 cuDNN SDK 7.6:
- 下载 cuDNN SDK 7.6 for CUDA® 10.1 安装文件并解压缩。
- 将解压后的文件夹中的文件拷贝到 CUDA® Toolkit 10.1 的安装目录下。例如,如果 CUDA® Toolkit 10.1 安装在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1,则将解压后的文件夹中的文件拷贝到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\。
4. 安装 TensorFlow-GPU 2.1.0:
- 打开命令提示符或 Anaconda Prompt,创建一个新的 Python 虚拟环境:`conda create -n tensorflow-gpu python=3.7`
- 激活虚拟环境:`conda activate tensorflow-gpu`
- 安装 TensorFlow-GPU 2.1.0:`pip install tensorflow-gpu==2.1.0`
注意:安装 TensorFlow-GPU 时需要保证 CUDA® Toolkit 和 cuDNN SDK 的版本与 TensorFlow-GPU 版本匹配,否则可能会出现兼容性问题。
阅读全文