非常详细的 Tensorflow GPU 版本安装教程
- 一、安装Anaconda
- 二、TensorFlow GPU
一、安装Anaconda
这一步比较简单,也没有太多的需要注意的,去官网下载即可:
官网地址如下:
https://www.anaconda.com/blog/individual-edition-2020-07/
二、TensorFlow GPU
为了实现 TensorFlow GPU 支持,需要各种驱动程序和库。
pip 软件包:
TensorFlow pip 软件包对采用 CUDA® 的显卡提供 GPU 支持:
pip install tensorflow
本指南将介绍最新稳定版 TensorFlow 的 GPU 支持和安装步骤。
旧版 TensorFlow:
对于 1.15 及更早版本,CPU 和 GPU 软件包是分开的:
pip install tensorflow==1.15 # CPU
pip install tensorflow-gpu==1.15 # GPU