Win10 Tensorflow-gpu安装配置

1.使用版本:

Python 3.6

CUDA 9.0

cuDNN (与CUDA版本对应)

tensorflow-gpu 1.8

注:可不安装anacondavs

2.下载CUDA并安装   https://developer.nvidia.com/cuda-toolkit-archive

(1)安装警告:The graphics driver could not find compatible graphics hardware。该错误可忽略。

(2)安装过程失败主要由于Visual Studio(VS) Intergration无法安装导致。选择自定义安装,不安装CUDA中的Visual Studio(VS) Intergration

(3)验证是否安装成功:命令行 nvcc -V


配置环境变量,将CUDA安装目录下的binlib\x64配置到path中。

3.下载解压cuDNN  https://developer.nvidia.com/rdp/cudnn-download

需要注册一下账号

一定要下载对应CUDA版本的cuDNN!!!

将解压后目录下的文件放置到CUDA安装目录下对应位置。lib放置lib目录下,以此类推。

4.安装tensorflow-gpu

(1)安装命令:

pip uninstall tensorflow(卸载cpu版本的tensorflow)

pip install tensorflow-gpu==1.8

如果出现错误:7201 (compatibility version 7200) but source was compiled with 7003 (compatibility version 7000). If using a binary install, upgrade your CuDNN library to match.  If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.类似,可能并不是cuDNN版本与CUDA不匹配,如果官网下的是匹配的,更大可能是由于tensorflow版本问题,可换更高版本试一试。

(2)测试是否安装成功:
import tensorflow as tf

test=tf.constant('hello gpu')

with tf.Session() as sess:

      print(sess.run(test))

检查gpu

tf.test.is_gpu_available()

#返回true/false

tf.test.gpu_device_name()

#设备名称

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值