Win10下Tensorflow-gpu2.0及Keras安装（基于anaconda）

zt3ff3n · 发表于 2022-6-24 09:28

总体思路是先安装CUDA和cuDNN，再安装Anaconda，构建tensorflow-gpu 2.0环境，安装Pycharm和Keras。
CUDA和cuDNN的下载地址：
CUDA
链接：https://pan.baidu.com/s/1dIXw--W4oqOMP0a1GS5UaA
提取码：la6m
cuDNN
链接：https://pan.baidu.com/s/10kEuuP6CtgxSTwXLuSEJDw
提取码：kjme

一、CUDA和cuDNN的安装及环境变量设置

首先，查看自己电脑显卡是否支持GPU，右键“此电脑”，点击管理进入“计算机管理”，点击“设备管理器”，查看“显示适配器”，我的是NVIDIA GeForce GTX960

在这个网站中找到对应的GPU，如果能找到，就继续安装GPU版本的tensorflow
https://developer.nvidia.com/zh-cn/cuda-gpus
如果支持，从以下网页下载tensorflow-gpu2.0对应的CUDA，我使用的是CUDA10.0，下载到本地之后双击进行安装。https://developer.nvidia.com/cuda-toolkit-archive

再从以下网站下载cuDNN，我是用的是7.5.0版本，解压后如图

复制到路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\cudnn

CUDA和cuDNN安装好后在环境变量中添加路径

二、安装Anaconda

官网下载链接https://www.anaconda.com/products/individual
我是用的是python3.7.4的版本
在这里查询以往的旧版本，下载到本地进行安装，这里贴一个比较靠谱的安装教程，不在此赘述
https://blog.csdn.net/ITLearnHall/article/details/81708148

三、tensorflow环境创建和安装

Anaconda安装好后，新建python环境用于安装tensorflow和Keras

#新建python环境
conda create -n TF_2G python==3.7.4

复制代码

根据提示输入“y”，继续进行安装，环境创建好后会出现

在此验证一下环境是否创建，输入

conda info --envs

复制代码

查看电脑中目前已有的环境，可以看到刚刚创建的环境TF_2G已存在

接下来是激活环境

conda activate TF_2G

复制代码

pip安装gpu版本的tensorflow2.0，这里可以参考其他教程改用清华源，这里我可以比较快的顺利安装，就没有换源。

pip install tensorflow-gpu==2.0.0-alpha

复制代码

出现这一句“Successfully installed...”就是安装好了，先不要关掉这个窗口，后面安装Keras还需要用到。下面先来测试一下tensorflow是否成功安装

四、Pycharm安装与Keras安装

首先安装一下Pycharm，社区版即可，版本我直接安装的官网的最新版，官网下载地址如下：
https://www.jetbrains.com/pycharm/download/#section=windows

安装过程可以参考这个博客，在此略过安装过程。
https://blog.csdn.net/Xuanze_xx/article/details/107926169跟着做到安装完成这一步就可以了。
pycharm安装好后，点击file—new project，出现新建工程的界面

选好location，点击红框最右边的“···”，选择我们刚创建的环境中的python.exe作为interpreter

创建好后，新建一个python.exe，写入如下代码

import tensorflow as tf
version = tf.__version__
gpu_test = tf.test.is_gpu_available()
print("tf version:",version,"\nGPU available",gpu_test)

复制代码

右键Run，说明之前安装的gpu版本的tensorflow2.0可用

下面安装对应版本的Keras
首先找到跟tensorflow2.0版本对应的Keras版本，是Keras2.3.1
https://docs.floydhub.com/guides/environments/

接着前面的anaconda prompt，在环境激活状态下，pip安装Keras2.3.1

pip install keras==2.3.1

复制代码

输入python，再输入import keras，没有报错就是安装成功了

所有的安装步骤就是这些了。

<div id="marketingBox" class="marketing-box"><div class="marketing-content">

CSDN 社区图书馆，开张营业！

深读计划，写书评领图书福利~

		自动登录	找回密码
密码			立即注册

Win10下Tensorflow-gpu2.0及Keras安装（基于anaconda）

本帖子中包含更多资源