您的位置:首页 > 其它

Tensorflow学习——Win10安装Tensorflow-gpu和CUDA及cuDNN

2018-06-08 23:13 706 查看

环境:Window10

Tensorflow安装官方链接:https://tensorflow.google.cn/install/install_windows

cuda安装官方链接:https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/

一、前期准备:

  1. 安装python3,下载网站:https://www.python.org/downloads/
  2. 安装numpy
  3. 检查电脑GPU是否达到要求,否则安装CPU版本,检查网站:https://developer.nvidia.com/cuda-gpus
  4. 安装Visual C++,若安装了Visual Studio则可不管这项


二、安装CUDA和cuDNN

    因Tensorflow-gpu对CUDA和cuDNN版本有所要求,根据安装的Tensorflow-gpu版本下载对应CUDA和cuDNN版本,可参照下图。

cuda下载链接:https://developer.nvidia.com/cuda-downloads

cudnn下载链接:https://developer.nvidia.com/rdp/form/cudnn-download-survey(需要注册填调查问卷)

下载后默认安装即可,安装完成后将目录\CUDA\v9.0\bin添加到系统路径Path中。
三、安装Tensorflow-gpu    本文采用pip的方式安装,打开Windows Powershell(建议打开管理员模式),运行下面命令,默认安装最新版本Tensorflow-gpu,若速度慢可以改用镜像。pip3 install --upgrade tensorflow-gpu
四、测试    打开Windows Powershell,输入下面命令。python

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
[/code]若获得以下结果则为安装成功。Hello, TensorFlow!

备注:

常见问题解决方法:https://www.tensorflow.org/install/install_windows#common_installation_problems

阅读更多
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: