1 准备工作
1.1 下载 ANACONDA并安装
链接:https://www.continuum.io/downloads
注意:安装时,这两个选项都勾一下,这样就不用在环境变量中手动添加anaconda了。
1.2 下载 CUDA并安装
链接:https://developer.nvidia.com/cuda-downloads
1.3 下载 CuDNN并解压
(需要注册,不过很方便)
链接:https://developer.nvidia.com/cudnn
2 配置环境变量
Anaconda、CUDA自动配好环境;
CuDNN下载下来是一个压缩文件,所以需要解压后手动配置环境。
打开解压后的文件,将有3个子文件夹。
打开“系统变量”中的“Path”,将上面3个子文件夹的位置添加进去。
3 安装tensorflow
非常简单。打开cmd,输入
pip install tensorflow-gpu
提示:如果出现pip版本低的情况,输入
python -m pip install -U pip
升级pip 即可。
等待安装完毕后,启动python,输入
import tensorflow
出现5个“successfully”,安装成功。