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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/magic_wz/article/details/79653628

安装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
以及操作系统版本
这里写图片描述

安装cudnn

下载地址:
https://developer.nvidia.com/cudnn
需要注册一下,才能下载,选择:
这里写图片描述
解压后把三个文件夹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

猜你喜欢

转载自blog.csdn.net/magic_wz/article/details/79653628