安装Tensorflow-gpu
Posted linjie1123
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了安装Tensorflow-gpu相关的知识,希望对你有一定的参考价值。
1. 下载nvidia驱动程序
下载后安装,全部默认,不用修改参数。
2. 选Anacodna Prompt
输入:conda install tensorflow-gpu
输入:y
#测试程序
import tensorflow as tf
sess = tf.Session()
a = tf.constant(10)
b= tf.constant(12)
sess.run(a+b)
结果为22
报错请更新h5py版本
以上是关于安装Tensorflow-gpu的主要内容,如果未能解决你的问题,请参考以下文章