【机器学习笔记】——TensorFlow的安装(cuda10.0+cuDNN7.4)

我是win7,显卡GTX1050Ti,选的cuda10.0,cuDNN7.4

tensorflow分为CPU版本和GPU版本,CPU版本运行较慢。

1 安装anaconda3(python3.6)

2 安装tensorflow

2.1 CPU版本安装

不依赖于其他,直接在cmd中 conda install tensflow

2.2 GPU版本安装

1)是否支持GPU版本

在控制面板-设备管理-显示配置器中查看是否有支持GPU的显卡

2)安装cuda

cuda最新(10.0)版本:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=7&target_type=exelocal

选择适合电脑系统win7的cuda安装包:

如果需要其他版本的cuda:https://developer.nvidia.com/cuda-toolkit-archive

安装步骤一步一步来就行。

3)安装cuDNN

cuDNN版本需要与cuda匹配

https://developer.nvidia.com/rdp/cudnn-download

如果需要其他版本的cuDNN:https://developer.nvidia.com/rdp/cudnn-archive?tdsourcetag=s_pctim_aiomsg

解压后,是这些:

将这三个复制至cuda的安装目录下就可以了。

4)安装tensorflow

conda install tensflow-gpu

你可能感兴趣的:(【机器学习笔记】——TensorFlow的安装(cuda10.0+cuDNN7.4))