TensorflowKeras下载安装

转自:https://zhuanlan.zhihu.com/p/95065951

以下是根据自己需要在上面教程基础上改动的

安装环境:

操作系统:win10

python版本:3.8

NVIDIA显卡驱动版本:432.0

tensorflow-gpub版本:2.3.1

CUDA:10.1

cudnn:8.0.3

keras版本:2.4.3

注意:

CUDA、cudnn、Python、tensorflow版本之间的匹配

激活tensorflow环境 ≠ 安装tensorflow

先安装tensorflow-gpu,再安装keras

一、       Tensorflow

TensorFlow是一个基于数据流编程(dataflow programming)的符号数学系统,被广泛应用于各类机器学习(machine learning)算法的编程实现,其前身是谷歌的神经网络算法库DistBelief
Tensorflow
拥有多层级结构,可部署于各类服务器、PC终端和网页并支持GPU和TPU高性能数值计算,被广泛应用于谷歌内部的产品开发和各领域的科学研究

TensorFlow
由谷歌人工智能团队谷歌大脑(Google Brain)开发和维护,拥有包括TensorFlow Hub、TensorFlow Lite、TensorFlow Research Cloud在内的多个项目以及各类应用程序接口(Application Programming Interface, API) [2]

自2015年11月9日起,TensorFlow依据阿帕奇授权协议(Apache 2.0 open source
license
)开放源代码

二、       Tensorflow-gpu

2.1   
Tensorflow-gpu下载

下载地址:https://pypi.org/project/tensorflow-gpu/#files

 如图:

 转载:Tensorflow、Keras下载安装

 

 

2.2   
tensorflow-gpu环境搭建

2.2.1 搭建tensorflow-gpu环境

打开cmd窗口,键入:

conda
create -n tensorflow-gpu python=3.8

2.2.2 激活tensorflow-gpu环境

在cmd窗口,键入:

activate
tensorflow-gpu

2.2.3 下载tensorflow

在cmd窗口,键入:

pip install tensorflow_gpu-2.3.1-cp38-cp38-win_amd64.whl

或(如果没有下载离线包)

 pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu

 

2.2.4 测试tensorflow-gpu安装完成

在cmd窗口,键入:

import tensorflow as tf

tf.compat.v1.disable_eager_execution()


hello = tf.constant('hello,TensorFlow!')
sess = tf.compat.v1.Session()
print(sess.run(hello))

三、      
安装keras框架

3.1   
安装keras

在cmd里激活tensorflow-gpu后,键入:

pip install -i
https://pypi.tuna.tsinghua.edu.cn/simple keras

 转载:Tensorflow、Keras下载安装

 

 

3.2   
测试keras安装完成

以下是MNIST手写数字数据集,如果能整成运行,说明keras安装完成。

 1 from keras.datasets import mnist
 2 from keras.utils import to_categorical
 3 
 4 train_X, train_y = mnist.load_data()[0]
 5 train_X = train_X.reshape(-1, 28, 28, 1)
 6 train_X = train_X.astype('float32')
 7 train_X /= 255
 8 train_y = to_categorical(train_y, 10)
 9 
10 from keras.models import Sequential
11 from keras.layers import Conv2D, MaxPool2D, Flatten, Dropout, Dense
12 from keras.losses import categorical_crossentropy
13 from keras.optimizers import Adadelta
14 
15 model = Sequential()
16 model.add(Conv2D(32, (5,5), activation='relu', input_shape=[28, 28, 1]))
17 model.add(Conv2D(64, (5,5), activation='relu'))
18 model.add(MaxPool2D(pool_size=(2,2)))
19 model.add(Flatten())
20 model.add(Dropout(0.5))
21 model.add(Dense(128, activation='relu'))
22 model.add(Dropout(0.5))
23 model.add(Dense(10, activation='softmax'))
24 
25 model.compile(loss=categorical_crossentropy,
26              optimizer=Adadelta(),
27              metrics=['accuracy'])
28 
29 batch_size = 100
30 epochs = 8
31 model.fit(train_X, train_y,
32          batch_size=batch_size,
33          epochs=epochs)
34 
35 test_X, test_y = mnist.load_data()[1]
36 test_X = test_X.reshape(-1, 28, 28, 1)
37 test_X = test_X.astype('float32')
38 test_X /= 255
39 test_y = to_categorical(test_y, 10)
40 loss, accuracy = model.evaluate(test_X, test_y, verbose=1)
41 print('loss:%.4f accuracy:%.4f' %(loss, accuracy))

 

 

 转载:Tensorflow、Keras下载安装