Windows10 下 利用 Anaconda3 +CUDA 8.0 + CUDNN 6.0 为 Pycharm 配置虚拟的 Tensorflow 环境 (保姆级教程)

为了本科图像处理的毕业设计,最近跟tensorflow的配置问题大战了几天几夜,终于搞定了
写这个贴子,是为了下次再配置的时候不用像这几天到处找教程,哭了
言归正传

前言

如果有需要用到tensorflow框架的,所有软件的安装版本一定要严格按照如下表格
(具体参考tensorflow各个版本的CUDA以及Cudnn版本对应关系
Windows10 下 利用 Anaconda3 +CUDA 8.0 + CUDNN 6.0 为 Pycharm 配置虚拟的 Tensorflow 环境 (保姆级教程)

如何选择版本

至于如何选择,取决于电脑的显卡,这里只介绍NIVDIA显卡的选择
进入到电脑的控制面板----->>NVIDIA控制面板
Windows10 下 利用 Anaconda3 +CUDA 8.0 + CUDNN 6.0 为 Pycharm 配置虚拟的 Tensorflow 环境 (保姆级教程)
再点击左下角的系统信息
Windows10 下 利用 Anaconda3 +CUDA 8.0 + CUDNN 6.0 为 Pycharm 配置虚拟的 Tensorflow 环境 (保姆级教程)
点击组件就可以看到啦!!!(3D设置的第三行 NVIDIA CUDA 8.0.0 driver)
Windows10 下 利用 Anaconda3 +CUDA 8.0 + CUDNN 6.0 为 Pycharm 配置虚拟的 Tensorflow 环境 (保姆级教程)
楼主的电脑是 NVIDIA 950M的显卡,所以这里选择:
CUDA 8.0
CUDNN 6.0
tensorflow-gpu 1.4.0 (tensorflow和tensorflow-gpu的区别在于,前者是CPU跑的,后者是GPU,GPU在做图像处理方面,后者领先前者很多个数量级)
可以先把CUDA 和 CUDNN 下载好备用,也可以下载完 Anaconda 和 tensorflow 再下载,楼主这里是先下载好了

CUDA 和 CUDNN 的下载

点击下方官网下载地址可以下载
NVIDIA官网下载CUDA (无需注册登录)
NVIDIA官网下载CUDNN(需要注册+登录)
下载速度可能比较慢,耐心等待~~ 也可以参考下述帖子
解决英伟达CUDA和cuDNN下载过慢的问题
cuda以及cudnn下载慢的问题解决!
注意CUDA的下载需要找到适合自己的版本,如下图,点击右下角的Legacy Releases,这里是旧版发布,在里面找到对应的版本下载即可CUDNN同理,需要对应的版本
Windows10 下 利用 Anaconda3 +CUDA 8.0 + CUDNN 6.0 为 Pycharm 配置虚拟的 Tensorflow 环境 (保姆级教程)
为了方便大家,可能有同样的配置的同学,这里吧CUDA 8.0 和CUDNN 6.0 放在百度云了,有需要的自提
百度云链接:CUDA + CUDNN (win10)
提取码:lrfx

CUDA 和 CUDNN的安装

参考下面的帖子,因为写这个的人太多了,我这里就没必要再贴图了
Win10安装CUDA10和cuDNN

待会继续更,未完待续…