【Deep Learning】win10下配置tensorflow1.4(gpu版本)
来源:互联网 发布:android答题系统源码 编辑:程序博客网 时间:2024/05/21 19:42
环境:
系统:win10
显卡:NVIDIA GTX960M
python:3.6
tensorflow:1.4
下载
下载CUDA8.0(必须8.0):https://developer.nvidia.com/cuda-80-ga2-download-archive
下载CUDNN6.0(必须6.0):https://developer.nvidia.com/rdp/cudnn-download
下载python3.6(必须3.0以上):https://www.python.org/downloads/
安装CUDA8.0
下载CUDA8.0后进行默认安装,等待完成后环境变量里会有:
CUDA_PATH CUDA_PATH_V8_0
两个环境变量。
命令行里输入:
nvcc -V //(大写的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
表明安装成功。
安装CUDNN6.0
下载CUDNN6.0,是一个压缩包,解压后会有三个文件夹bin,lib和include。将这三个文件夹下的内容复制到CUDA8.0的安装目录下的对应文件夹下即可。
如:
将bin下的cudnn64_6.dll复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下的bin文件夹下。
安装python3.6
下载python3.6后直接安装,完成后配置环境变量如:
//python.exe所在目录C:\Program Files\Python36//pip3.exe所在目录C:\Program Files\Python36\Scripts
在命令行输入:
python -V //(大写的V)
输出:
Python 3.6.3
表明安装成功
安装tensorflow1.4
打开命令行(必须以管理员身份运行),输入
pip3 install tensorflow-gpu
等待下载安装完成(约67M)。
下载速度可能比较慢,所以可以更换为阿里云的镜像
pip3 install tensorflow-gpu -i https://mirrors.aliyun.com/pypi/simple/
这样下载速度会提高很多。
安装完成后,进入python命令行测试。输入:
tf.Session()
输出:
2017-12-08 17:47:08.439868: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\36\tensorflow\core\common_runtime\gpu\gpu_device.cc:1120] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce GTX 960M, pci bus id: 0000:01:00.0, compute capability: 5.0) <tensorflow.python.client.session.Session object at 0x000002887924D908>
表明配置成功。(识别的出显卡信息:device: 0, name: GeForce GTX 960M, pci bus id: 0000:01:00.0, compute capability: 5.0)
vscode下测试
首先配置python运行环境(省略),然后新建测试文件test.py
输入:
import tensorflow as tfwith tf.device('/gpu:0'): a = tf.constant([1.0, 2.0, 3.0]) b = tf.constant([1.0, 2.0, 3.0])sess = tf.Session()print(sess.run(a+b))
输出:
[ 2. 4. 6.]
整个环境配置成功。
- 【Deep Learning】win10下配置tensorflow1.4(gpu版本)
- 【Deep Learning】win10下无GPU配置tensorflow
- win10下配置Tensorflow1.4
- Win10 配置tensorflow1.3 GPU版本以及tensorboard的使用
- Win10 TensorFlow1.4 (gpu)安装
- win10下安装TensorFlow1.4.0-GPU
- Ubuntu16.04 配置tensorflow1.0 gpu版本
- Ubuntu16.04 配置tensorflow1.0 gpu版本
- keras配置(WIN10/GPU版本)
- Win10下安装TensorFlow(GPU版本)
- Deep Learning: win7安装Theano,配置GPU
- Deep Learning---win10 搭建caffe开发环境(GPU)
- Win10下安装GPU版本的tensorflow
- win10下安装tensorflow1.0
- 【TFBOYS教程】Win10下基于Anaconda3的TensorFlow1.4安装(避坑小指南)
- win10装anaconda3-4.2.0、python3.5、opencv3.0和tensorflow1.2(gpu)
- Tensorflow1.4安装gpu教程
- keras deep learning use gpu
- Mahout推荐算法API详解
- linux网络驱动分析
- 【Person Re-ID】SVDNet for Pedestrian Retrieval
- 学徒浅析Android开发——SO文件的混淆
- 基于IC设计的实用ISP介绍
- 【Deep Learning】win10下配置tensorflow1.4(gpu版本)
- c++记录
- 1023. 组个最小数 (20)
- mysql分库分表
- 《第一本Docker书》读书总结(一)第4章
- WIFI开发版实验,ssh网络服务开发板访问虚拟机ubuntu文件中,ssh连接ubuntu时ssh: Exited: Error connecting: Connection refused
- python 安装urllib
- Linux下tp5找不到模板文件解决方法
- LVS+KEEPALIVED高可用群集实验分步配置简要