cuda out of memory gpu还有空间_windows10下Tensorflow2.0-gpu配置（极简主义）

第一步：下载CUDA Toolkit 10.0（tensorflow2.0）

下载好之后安装

画面1

画面2.

画面

画面3.，这里你同意就好了

画面4，这里你选高级

画面

画面5，把那个√去掉，以免出错

这里我选默认，下面照这个说明

这里

安装好之后把cudnn下好

https://developer.nvidia.com/rdp/cudnn-download​developer.nvidia.com

没有的话先注册，注册成功之后再来点击这个，不然很麻烦，晕头转向的

打好沟，选择windows10

下好之后解压，里面有三个文件

把这真个文件夹都复制到CUDA的那个安装文件目录，我这是C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0

大概就是这个意思

下面配置环境

网上查查找找，反正我是配置了五个，凑合用，勿喷哈哈哈

然后就是安装tensorflow-gpu==2.0.0,

pip install tensorflow-gpu==2.0.0

我想想，还有什么

如果GPU显示True就没啥问题

把显存限制一下，不然容易爆

import tensorflow as tf
tf.config.gpu.set_per_process_memory_growth(enabled=True)

嗯...差不多了，这样子在我电脑什么是没有问题

本人RTX 1660ti

3384张224x224x3的图片训练大概跑比之前快了十多倍吧，勉勉强强

记得点个赞点个关注哦