用GPU训练神经网络的设置步骤

1. 安装Anaconda

2. 在Anaconda 的 prompt 中新建一个环境： conda create --name TENSORFLOW-GPU python=3.9

3. 安装cuda 和cudnn (版本要对应,我用的是 cuda = 11.6, cudnn=8.4）

cuda 装好后，要把cudnn中得文件粘贴到cuda对应的文件夹中（相当于补丁）；另外还需要把这几个文件夹在的位置放在环境变量中

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

 

4. 在 Anaconda prompt 中执行下列操作：

•   4.1 打开新建的环境： activate TENSORFLOW-GPU

•   4.2 下载TensorFlow-gpu: pip install tensorflow-gpu==2.8.0 (与cuda 和cudnn版本对应）

•    4.3 检查GPU 是否被调用 tf.config.list_physical_devi