最新Window10版本下tensorflow-gpu版安装详细步骤
一、首先确定你的显卡对应的cuda版本
桌面上鼠标右键选择NVIDIA控制面板,左下角选择系统信息后选择组件,看你的电脑应该安装哪个版本的cuda
二、可以去官网下载你所对应的版本
如果你是cuda10.0可以直接下载下面百度云链接
链接:https://pan.baidu.com/s/174YZ5JUERqBxeAqnCedRCQ
提取码:303p
三、cuda安装步骤
cuda10.0选择自定义安装,我的路径是C:Users\陈超\ ,在安装过程中其余选择路径时均可以按默认路径安装。安装完成后就需要配置环境了,在设置中搜索环境变量,接下来点击环境变量——双击Path——新建路径(见下图,需要建立两个路径)——点击确定
四、下载cudnn
cudnn下载后可以解压在任意目录下,解压后需要把cudnn目录里各个文件夹里面的文件分别对应复制到cuda文件夹里
五、安装tensorflow
如果你是下载上面百度云提供的tensorflow版本,那么可以直接把tensorflow-1.12.0rc0-cp36-cp36m-win_amd64.whl 这个文件放到annconda中的scripts文件夹中 我的路径是 D:\Anaconda\Scripts
打开Anaconda Prompt,按以下步骤进行
等待一会应该就安装好了
六、测试tensorflow成功并且是否是gpu版本
import tensorflow as tf #没报错即安装成功
#测试是否是gpu版本
import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
if __name__ == "__main__":
print(device_lib.list_local_devices())
当输出是GPU时即说明gpu版本安装成功