Python下使用TensorFlow测试MNIST数据神经网络入门。

最新推荐文章于 2024-08-08 16:44:55 发布

Small__Two

最新推荐文章于 2024-08-08 16:44:55 发布

阅读量3.2k

点赞数 3

分类专栏：神经网络实战文章标签： Python TensorFlow Cuda MNIST

本文链接：https://blog.csdn.net/u014072439/article/details/79038463

版权

本文介绍了如何在Python3.5环境中，利用TensorFlow1.4和CUDA8.0搭建神经网络，对MNIST数据集进行测试。首先，需要将MNIST数据下载并放在/tmp/目录下，然后通过提供的代码运行实现测试，并展示测试结果。

摘要由CSDN通过智能技术生成

参考教程：http://blog.topspeedsnail.com/archives/10377

环境：

python3.5

cuda8.0

TensorFlow1.4

先下载MNIST测试数据放至/tmp/目录。MNIST数据下载http://yann.lecun.com/exdb/mnist/（直接下载不了的话，获取链接地址放迅雷上下载即可）

代码如下，根据参考教程修改了落后的版本语句部分。

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2018/1/11 20:37
# @Author  : lj
# @Site    : 
# @File    : hasTensorFlowPart1.py
# @Software: PyCharm Community Edition

import tensorflow as tf
import numpy as np

# tensorflow自带了MNIST数据集
from tensorflow.examples.tutorials.mnist import input_data

# 下载mnist数据集
mnist = input_data.read_data_sets('/tmp/', one_hot=True)
# 数字(label