记录win10安装keras & tensorflow-gpu过程

安装环境

win10 64bit

python 3.6.12

CUDA &cuDNN根据自己电脑情况下载安装。

创建虚拟环境:

1、我这里创建的是python36,创建完成后,激活环境,打开Anaconda prompt,输入conda python36激活环境。
输入

pip install --ignore-installed --upgrade tensorflow-gpu

在这里插入图片描述

验证tensorflow-gpu:

进入python
在这里插入图片描述
输入:

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

在python36环境中输入:

pip install keras 

等待安装完成后进入python中,输入import keras 即可。

纯属记录学习过程,如有侵权请联系我。

猜你喜欢

转载自blog.csdn.net/qq_38233472/article/details/109290526
今日推荐