最新Window10版本下tensorflow-gpu版安装详细步骤

一、首先确定你的显卡对应的cuda版本

桌面上鼠标右键选择NVIDIA控制面板,左下角选择系统信息后选择组件,看你的电脑应该安装哪个版本的cuda

最新Window10版本下tensorflow-gpu版安装详细步骤_第1张图片

二、可以去官网下载你所对应的版本

如果你是cuda10.0可以直接下载下面百度云链接

链接:https://pan.baidu.com/s/174YZ5JUERqBxeAqnCedRCQ 
提取码:303p 


三、cuda安装步骤

cuda10.0选择自定义安装,我的路径是C:Users\陈超\   ,在安装过程中其余选择路径时均可以按默认路径安装。安装完成后就需要配置环境了,在设置中搜索环境变量,接下来点击环境变量——双击Path——新建路径(见下图,需要建立两个路径)——点击确定

最新Window10版本下tensorflow-gpu版安装详细步骤_第2张图片

 

四、下载cudnn

cudnn下载后可以解压在任意目录下,解压后需要把cudnn目录里各个文件夹里面的文件分别对应复制到cuda文件夹里

五、安装tensorflow

如果你是下载上面百度云提供的tensorflow版本,那么可以直接把tensorflow-1.12.0rc0-cp36-cp36m-win_amd64.whl 这个文件放到annconda中的scripts文件夹中  我的路径是 D:\Anaconda\Scripts

打开Anaconda Prompt,按以下步骤进行

最新Window10版本下tensorflow-gpu版安装详细步骤_第3张图片

等待一会应该就安装好了

六、测试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())

最新Window10版本下tensorflow-gpu版安装详细步骤_第4张图片

当输出是GPU时即说明gpu版本安装成功 

 

 

你可能感兴趣的:(最新Window10版本下tensorflow-gpu版安装详细步骤)