1.在虚拟环境下下载cuda和cudnn,并安装tensorflow-gpu。参考以下博文,可行。
多环境安装,虚拟环境中的Tensorflow-gpu以及cuda及cudnn安装Windows_打酱油的劫少博客-优快云博客
2.直接用conda install tensorflow-gpu=1.12.0,同样可行,并且无需自行下载cuda和cudnn。conda会直接下载当前适用版本的cuda和cudnn。可行
参考来自(页面最下方)Python 工具 之 tensorflow gpu 环境的搭建配置(必要的相关工具 cuda\cudnn等 的下载安装)的详细步骤_仙魁XAN-优快云博客
3.测试是否安装成功gpu版本。会输出gpu信息。
python
import tensorflow as tf
sess=tf.Session(config=tf.ConfigProto(log_device_placement=True))