配置这套环境着实是废了一些劲,留个备份
本流程是在Windows10系统,进行Tensorflow+CUDA9.0的配置,其中使用Anaconda3来进行环境管理,Python版本为3.6。具体流程如下:
1.下载相关资源
CUDA9.0:https://developer.nvidia.com/cuda-80-ga2-download-archive,即cuda_9.0.176_win10.exe;
CUDNNv7.0:https://developer.nvidia.com/cudnn,进入后选择版本:cudnn-9.0-windows10-x64-v7.zip;
Anaconda3:https://www.anaconda.com/download/,选择Python3.6对应版本,当前最新版为5.0.1:Anaconda3-5.0.1-Windows-x86_64.exe;
2.安装CUDA9.0、CUDNN7.0和Anaconda3
CUDA8.0和CUDNN6.0默认安装即可,在安装Anaconda时,要勾选“加入环境变量”和“默认Python3.6”(后者默认勾选),其他默认即可,安装位置可以更改。
3.配置Anaconda
在开始菜单找到并打开AnacondaPrompt,可以看到目前处于Anaconda3环境(也就是“根目录”环境)下,在打开的命令后输入:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
4.建立并激活Tensorflow环境
继续在命令行输入:
conda create --name=tensorflow python=3.6,这句话十分好,不管你anaconda安装的是什么版本,这里创建的环境都是按照你设定的python版本安装的,十分便于安装多版本python,这里即便创建一个python2.7的环境也没有问题
这里,由于在Anaconda3安装的时候,选择了默认Python3.6版本,因此新建立的tensorflow环境就是基于Python3.6。再输入:
activate tensorflow (如果是ubuntu环境,此处为source activate tensorflow)
即可激活tensorflow环境,用deactivate可以取消激活。
5.安装Tensorflow
在激活tensorflow环境状态下,输入:
pip install tensorflow_gpu
会自动选择目前最新的1.5.0版本(支持CUDA8.0和Python3.6)进行安装。
6.测试Tensorflow
在激活tensorflow环境状态下,进入python,输入:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
如果正常输出Hello, TensorFlow!,则为安装成功。