机器学习keras环境配置(WINDOW10 + NV的GPU)

安装Anaconda3

下载地址:
https://repo.continuum.io/archive/Anaconda3-5.1.0-Windows-x86_64.exe
使用Python3的64位版本

安装C++环境

下载地址:
https://www.microsoft.com/en-us/download/details.aspx?id=53587

安装CUDA

下载地址:
https://developer.nvidia.com/cuda-80-ga2-download-archive
选择CUDA Toolkit 8.0
以及操作系统版本
机器学习keras环境配置(WINDOW10 + NV的GPU)_第1张图片

安装cudnn

下载地址:
https://developer.nvidia.com/cudnn
需要注册一下,才能下载,选择:
机器学习keras环境配置(WINDOW10 + NV的GPU)_第2张图片
解压后把三个文件夹COPY到CUDA安装目录下,如图:

安装PYTHON依赖环境

以管理员权限启动conda的命令行界面:

conda install opencv=3.3.1
conda install tensorflow-gpu
conda install keras

验证

打开Python交互,输入:

import tensorflow as tf
hello = tf.constant("Hello, TensorFlow!")
sess = tf.Session()

测试是否可运行

参考:
https://zhuanlan.zhihu.com/p/29841665

你可能感兴趣的:(机器学习)