[Python] Tensorflow with GPU

使用Python 做机器学习/深度学习时常使用的Tensorflow
还有分成CPU版本与GPU版本。
常常安装时容易在小细节碰壁,今天就把安装过程简单纪录。

建立对应的Python版本的虚拟环境

也因为各个tensorflow版本需要搭配的python版本不同,建议在安装之前,依据所需要的pyhton 版本建立虚拟环境(虚拟环境建立方法)。
先确认tensorflow.org支援对应的版本,
以tensorflow 1.12版为例,所对应的python是3.5-3.6。
http://img2.58codes.com/2024/20115086gjWcPVUhAa.png

安装Cuda / cuDNN

GPU版本的tensorflow,还需要安装 nvida Cuda & cnDNN,这部分也是比较容易犯错的部分

Cuda(Compute Unified Device Architecture)

NVIDIA所推出的一种整合并行计算框架

Cudnn(CUDA Deep Neural Network library)

NVIDIA打造的针对深度学习网路的加速library

以tensorflow 1.12版为例,所对应的Cuda cuda9.0 / cuDNN7.2+。
http://img2.58codes.com/2024/20115086CJPgTWpqGB.png

下载Cuda要注意下载对应的版本,这次需要的是9.0版,下载后直接安装后记得重开机。
Cuda

紧接着是下载cuDNN,注意7.2 fow windows
(20210414更新 : 如果是比较旧的版本会在这里cuDNN)
下载解压缩后建议放在C:\Program Files\NVIDIA GPU Computing Toolkit下。
hcuDNN

解压缩后将相关资料夹位置加入PATH环境变数

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin#(有些版本是libx64有些是lib64 要道对应的资料夹确认)C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\extras\CUPTI\libx64 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\include#(依据刚刚解压缩的cuDNN资料夹位置)C:\Program Files\NVIDIA GPU Computing Toolkit\cudnn-10.0-windows10-x64-v7.2 

安装tensorflow

开启相对应环境的prompt,利用pip指令就可以快速安装
tensorflow 2.X 开始CPU和GPU版本用相同套件直接pip对应版本就可以

pip install tensorflow==2.3

tensorflow 1.X 版本CPU与GPU需安装不同套件

# CPU versionpip install tensorflow==1.X# GPU versionpip install tensorflow-gpu==1.X

关于作者: 网站小编

码农网专注IT技术教程资源分享平台,学习资源下载网站,58码农网包含计算机技术、网站程序源码下载、编程技术论坛、互联网资源下载等产品服务,提供原创、优质、完整内容的专业码农交流分享平台。

热门文章