> 文章列表 > Tensorflow GPU 版本安装教程

Tensorflow GPU 版本安装教程

Tensorflow GPU 版本安装教程

非常详细的 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

Tensorflow GPU 版本安装教程

香烟网