跳转至

Keras 安装配置

1 安装 Miniconda / Anaconda

  书接上回:Win11 安装配置 Miniconda

2 创建并激活虚拟环境

2.1 创建虚拟环境

创建名为 krs01 的 python3.8 虚拟环境

conda create -n krs01 python=3.8

2.2 激活虚拟环境

激活虚拟环境

conda activate krs01

3 安装并运行 Jupyter Notebook

3.1 安装 ipython 和 notebook

使用 conda 在对应的虚拟环境中安装

conda install -n krs01 ipython notebook

3.2 运行 Jupyter Notebook

运行 jupyter notebook

jupyter notebook
  • vscode 终端界面

  • 浏览器运行界面

  • 注:按 Ctrl + C 关闭 Notebook

4 配置 Keras 环境

4.1 安装主要依赖包

conda install -n krs01 numpy matplotlib pandas pydotplus h5py scikit-learn

conda install -n krs01 scipy mkl-service libpython m2w64-toolchain

4.2 安装 TensorFlow

安装 TensorFlow

pip install --ignore-installed --upgrade tensorflow==2.10.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 在安装 TensorFlow 时,Keras 也会随之安装,不用单独安装,且当前的版本中,TensorFlow 会同时安装 CPU 和 GPU 两个版本。
  • --ignore-installed --upgrade :忽略已安装的升级
  • -i https://pypi.tuna.tsinghua.edu.cn/simple:切换清华源

https://www.tensorflow.org/install/source_windows#gpu

用 conda 安装 CUDA、cuDNN

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
  • 这里我们指定安装与 tensorflow 2.10.0 匹配的 CUDA11.2cuDNN 8.1.0

image-20230428092112240

5 安装 CUDA

5.1 检查当前版本

检查当前 nvidia 显卡版本指令

nvidia-smi
  • 此时可以看到当前显卡驱动为 511.69,支持的最高 CUDA 版本为 11.6

5.2 官网下载

image-20230428201253558

5.3 安装注意

  与普通软件安装不同,在安装 CUDA 时,需要设置两次。第一次要求设置临时解压目录,第二次才是设置安装目录,这里建议默认即可,也可以自定义。安装结束后,临时解压文件夹会自动删除。切记,使用两次设置的路径不能相同,否则安装后会找不到目录。同时,因为是系统级软件,所以最好全程使用默认的安装位置。

  • 解压 CUDA 安装包

image-20220726110749394

  • 选择自定义安装

image-20230428193829668

  • 取消安装 Visual Studio Integration 组件,防止后续报错

image-20230428194413913

  • 使用默认安装路径,不做修改

image-20230428194440733

检查是否安装成功

  • 检查当前 CUDA 版本指令
nvcc -V
  • 输出版本即为成功

image-20230428203435904

6 安装 cuDNN

6.1 官网下载

  • 官网:cuDNN 下载,这里 必须登录账号 才能继续下载。

image-20220726135042391

  • 首次使用需要注册账号并输入详细信息

image-20220726140602291

6.2 下载对应版本

  cuDNN下载,因为之前安装的 CUDA 版本号为 11.2,所以我们选择对应版本的 cuDNN 进行下载。

image-20230428201412685

6.3 安装 cuDNN

  上面我们下载的内容实际上是压缩包文件,本质上就是对 CUDA 的关于深度学习的补充文件,我们需要将解压后的文件内容找到 CUDA Development 安装目录中对应目录,打开文件夹 逐一复制内部内容。需要注意的是:

lib 相关文件需要复制到 lib\x64 目录中。

1、cuDNN8.1.0 解压包内容

image-20230428202404924

2、CUDA Development 安装目录内容

image-20220726145523088

3、逐个复制内容到对应目录中

image-20230428202655263

7 配置环境变量

  在成功安装 CUDA 后,系统会自动添加环境变量

image-20230428201014850

在环境变量中添加如下内容

  如若在系统环境变量中没有,则需要自行添加如下内容。【注:此处安装目录以实际情况为准】

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\libnvvp

image-20230428201050040

8 测试安装环境

import scipy
import numpy
import matplotlib
import pandas
import sklearn
import pydotplus
import h5py

import theano
import tensorflow
import keras

print('scipy ' + scipy.__version__)
print('numpy ' + numpy.__version__)
print('matplotlib ' + matplotlib.__version__)
print('pandas ' + pandas.__version__)
print('sklearn ' + sklearn.__version__)
print('h5py ' + h5py.__version__)

print('theano ' + theano.__version__)
print('tensorflow ' + tensorflow.__version__)
print('keras ' + keras.__version__)

参考文章

  1. Build from source on Windows | TensorFlow
  2. CSDN - CUDA安装
  3. CSDN - CUDA安装教程(超详细)
  4. CSDN - 深度学习环境搭建超级详解(Miniconda、pytorch安装)
  5. CSDN - TensorFlow 2.8.0安装 + Miniconda + GPU支持
  6. 知乎 - 深度学习GPU环境CUDA详细安装过程(简单快速有效)