tensorflow安装说明
平台:目前可在Ubuntu、Mac OS、Windows上安装
版本:提供gpu版本、cpu版本
安装方式:pip方式、Anaconda方式
Tips:
- 在Windows上目前支持python3.5.x
- 检测目前安装了哪些环境:安装环境需要cuda8,cudnn5.1
具备了上述安装条件(CPU或GPU)之后,TensorFlow可以通过两种方式进行安装。一是”native” pip,
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple/ --upgrade tensorflow # for Python 3.*
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple/ --upgrade tensorflow-gpu # for Python 3.* and GPU
二是Anaconda。
本文中是使用Anaconda安装的,所以详细介绍一下Anaconda安装过程。
1. 安装CUDA® Toolkit 8.0+cuDNN v5.1
1)CUDA8.0安装
下载后直接默认安装,中间时间有点长,大家耐心等待。(若在检测显卡版本时报错,提示你的显卡驱动太新,这就是cuda的版本低了,安装一个高一些的版本,注意cudnn版本也要对应)。
默认安装基本不会出错,路径cuda会自动加入系统环境变量中。
检查是否装好cuda方法:终端输入:nvcc -V 出来正确的版本就是装好了。
2)CuDNN5.1安装
载到C盘根目录,解压到C盘根目录,解压后默认名字为cuda,不要改!!!。然后将cudnn的bin目录加入Path环境中:C:\cuda\bin。同时在系统变量中新建一个cudnn的环境变量路径,(方法:设置→控制面板→系统→查看高级系统设置→环境变量)然后
将<installpath>\cuda\bin\cudnn64_5.dll 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
将<installpath>\cuda\ include\cudnn.h 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
将<installpath>\cuda\lib\x64\cudnn.lib 复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
2.安装Anaconda+tensorflow
1) 下载安装Python 3.5.x 64-bit,要注意版本3.5.x。
2) 下载安装Anaconda;
我下载的是Anaconda4.3.0For Windows 64bit(内置python3.6),下载好了就安装,一直下一步。
1.检查Anaconda是否成功安装:
conda --version
2.检测目前安装了哪些环境:
conda info --envs
3.检查目前有哪些版本的python可以安装:
conda search --full-name python
4.安装不同版本的python并创建tensorflow-gpu环境:打开Anaconda Prompt,进入Anaconda命令行管理界面。配置清华仓库镜,输入指令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
创建运行环境,输入指令:
conda create -n tensorflow-gpu python=3.6
(n=name)新建一个名字叫“tensorflow-gpu”,python版本为3.6的运行环境,此环境与Anaconda中其它环境隔离。红框中的软件包也会随之安装,输入“y“和回车后开始安装。
5.按照提示,激活之:</