安装tensorflow-gpu

pip install --upgrade tensorflow-gpu

import tensorflow as tf
sess = tf.Session() 

如果提示如下

-- ::51.668966: I tensorflow/core/platform/cpu_feature_guard.cc:] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA

解决方法

https://www.cnblogs.com/sea-stream/p/10432698.html

参考：

https://blog.csdn.net/u014610925/article/details/79642076

https://blog.csdn.net/haoyu_does/article/details/79779773