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 即可。