1. 首页
  2. 编程语言
  3. Python
  4. Tensorflow设置显存自适应,显存比例的操作

Tensorflow设置显存自适应,显存比例的操作

上传者: 2021-05-03 00:50:07上传 PDF文件 46.81 KB 热度 13次

Tensorfow框架下,在模型运行时,设置对显存的占用。设置GPU的使用率的时候,都是在创建Session的时候,对config类进行设置。此外,当电脑上有多块GPU的时候,可以指定选取哪一快GPU进行计算。

下载地址
用户评论