WIN10+tensorflow+ananconda+cuda9.0+cudnn7.0环境配置记录

配置这套环境着实是废了一些劲,留个备份

本流程是在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!,则为安装成功。


转一篇linux环境的配置https://blog.csdn.net/ZWX2445205419/article/details/69429518

你可能感兴趣的:(tensorflow,cuda,environment)