win10 安装tensorflow

来源:互联网 发布:矩阵特征值分解 编辑:程序博客网 时间:2024/05/18 13:24


1、cuda toolkit V8.0安装,可以去官网下载,
cuda_8.0.61_win10
注意1:安装目录别改,默认在c盘 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
注意2:安装错了可以重装

2、cuDNN拷贝到cuda安装目录:
去官网注册,然后下载:cudnn-8.0-windows10-x64-v5.1
即使你把cuda安装在d盘,也会在这个目录
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

安装完毕后测试cuda版本:
nvcc -V
结果:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Mon_Jan__9_17:32:33_CST_2017
Cuda compilation tools, release 8.0, V8.0.60
然而这个只是cuda的安装结果,cudnn安装成功后,tensorflow能检测出gpu,cudnn不成功的话,tensorflow不能检测出gpu
3、卸载tensorflow cpu版    
1)如果用conda安装:
conda install tensorflow
如果用pip安装:
pip install tensorflow
4、安装gpu版(conda无安装包)
pip install tensorflow-gpu
5、验证至看到显卡信息则成功
  python环境下:
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

4.通过Pip安装tensorflow

谷歌官网上说的比较清楚,如何通过pip安装tensorflow的CPU版本和GPU版本。但是按照官网上的操作会出现一些问题。

举例:假如你打算安装GPU版本,则只需要在命令行输入如下命令:

C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl
但如果只输入这些,会报出 
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl
cannot remove nonexistent setuptools的错误。

正确的做法:

C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl
pip install --upgrade --ignore-installed https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow-0.12.0rc0-cp35-cp35m-win_amd64.whl
一定要加上‘--ignore-installed’。
所以第四步总的来说就是在你的命令行里输入上面的代码。
5.在命令行里试运行:

可以看到所有库都加载成功,没有第三步里出现的问题。



参考:http://blog.csdn.net/joker316701882/article/details/54426004