windows10 + 64位python3.6.5 + tensorflow-gpu + GTX 950M

windows10 + 64位python3.6.5 + tensorflow-gpu + GTX 950M

这是我写的第一篇文章,大家多多包涵

  1. 由于tensorflowcpu版本实在太慢,cpu太烫。狠下心, 最后终于安装好gpu版本,大家可以参考一下,希望大家可以少踩坑。
  2. 这个过程python3.5也可以参考,过程差不多的,前提必须是64位的。
  3. 话不多说先上成功的图
    在这里插入图片描述

注意点

  • tensorflow-gpu版本是有对应的CUDA的,必须下载对。可以先百度找找要安装的版本对应的CUDA
  • tensorflow-gpu1.9.0 对应的CUDA版本为V9.0

下面说正事


1.检查显卡型号是否支持cuda tookit,可在NVIDIA的网站上查询 https://developer.nvidia.com/cuda-gpus

在这里插入图片描述

2. 安装tensorflow-gpu包时,检查核对tensorflow-gpu版本所要求的CUDA版本号,否则依然回报错误异常

本次安装的tensorflow-gpu1.9.0 对应的CUDA版本为V9.0 ,下载:https://developer.nvidia.com/cuda-toolkit-archive

以及下载CUDAV9.0对应的CUDNN版本(须注册下载): https://developer.nvidia.com/rdp/cudnn-download

下载后就是安装CUDA,双击安装默认那些就好

安装好后就是在系统环境变量里面添加(默认安装的话就是这个路径)
也可以自己右键查看安装的CUDA文件所在路径
在这里插入图片描述

打开文件所在的路径, 将对应的CUDNN解压到的内容直接复制到CUDA的文件目录中

在这里插入图片描述

  • 我的就是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

在这里插入图片描述

完成。测试代码,没报错就好了

import tensorflow as tf

a = tf.constant(0.1, dtype=tf.float32)

with tf.Session() as sess:
	sess.run(a)

猜你喜欢

转载自blog.csdn.net/weixin_43741442/article/details/84260575