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/
一、前期准备:
- 安装python3,下载网站:https://www.python.org/downloads/
- 安装numpy
- 检查电脑GPU是否达到要求,否则安装CPU版本,检查网站:https://developer.nvidia.com/cuda-gpus
- 安装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
阅读更多相关文章推荐
- Win10 Anaconda下TensorFlow-GPU环境搭建详细教程(包含CUDA+cuDNN安装过程)
- Ubuntu16.04 安装 CUDA8.0 + cudnn5.1 + TensorFlow(GPU) 详细过程
- Ubuntu16.04+cuda-8.0+cudnn-v5.1+tensorflow0.8-gpu/tensorflow1.0-gpu安装教程
- win10 配置tensorflow(GPU) anaconda3 cuda9.0 cudnn for 9.0
- ubuntu 16.04 安装 tensorflow-gpu 包括 CUDA ,CUDNN,CONDA
- Py之TF/Cuda/Cudnn:Win10下安装深度学习框架Tensorflow+Cuda+Cudnn最简单最快捷最详细攻略—Jason niu
- ubuntu16.04下安装CUDA cuDNN及tensorflow-gpu版本及caffe-gpu过程
- TensorFlow_gpu版本 python库安装流程(包括驱动,cuda,cudnn)
- ubuntu16.04安装NIVIDIA显卡驱动,cuda8.0,cuDNN6.0以及基于Anaconda安装Tensorflow-GPU
- Ubuntu安装Tensorflow GPU 版本和 CUDA Toolkit 9.1 和 cuDNN 7.0.5 for Python 3
- Ubuntu 16.04lts 安装NVIDIA 私有驱动、cuda、cudnn、tensorflow-gpu等问题
- Ubuntu环境下深度学习cuda,cudnn,caffe,tensorflow的安装
- Ubuntu16.04安装gpu版tensorflow1.2+cuda+cudnn
- ubuntu 16.04 安装 tensorflow-gpu 包括 CUDA ,CUDNN,CONDA
- Ubuntu 14.04 安装 CUDA8.0 cudnn 5.1 tensorflow1.2.1GPU
- windows安装tensorflow简单直接的方法(win10+pycharm+tensorflow-gpu1.7+cuda9.1+cudnn7.1)
- win10+cuda8.0+cudnn+Tensorflow(GPU)安装
- win10 安装cuda9.0+cuDNN7+tensorflow-gpu1.4
- 配置深度学习GPU加速(Cuda以及Cudnn安装,win10操作系统下)
- ubuntu下安装cuda,cudnn以及tensorflow(gpu)