Tensorflow框架中GPU的使用与否不是只会影响速度吗,为什么还会影响运行结果?

发布于 09-07 21:32 字数 1398 浏览 27 评论 0

Tensorflow框架中GPU的使用与否不是只会影响速度吗,为什么还会影响运行结果。。

关闭GPU后得到的结果会是一片空白。。。使用GPU则正常

https://pan.baidu.com/s/1mfA7...

代码如上,可以直接运行的

主要代码如下:

# -*- coding: utf-8 -*-
import numpy as np
from scipy.misc import imread, imsave
import argparse
import time

import os

# os.environ["CUDA_VISIBLE_DEVICES"] = "-1"  # 参数为 -1 表示只使用 CPU ,注释后可启用GPU加速
import tensorflow as tf


def run(session):
    args = argparse.Namespace()
    args.input = './7.jpg'
    args.output = './7_666.jpg'
    args.model = './models/wave.model'
    args.arch = "./models/model.meta"
    args.image = imread(args.input, mode="RGB").astype(np.float32)
    args.image = np.expand_dims(args.image, axis=0)

    # 使用模型
    saver = tf.train.import_meta_graph(args.arch, clear_devices=True)
    saver.restore(session, args.model)
    inputs = tf.get_collection("inputs")[0]
    output = tf.get_collection("output")[0]
    result = output.eval({inputs: args.image})
    result = np.clip(result, 0.0, 255.0).astype(np.uint8)
    result = np.squeeze(result, 0)
    imsave(args.output, result)


def main():
    session = tf.Session()
    with session.as_default():
        run(session)


if __name__ == "__main__":
    time_a = time.time()
    main()
    time_b = time.time()
    print('run_time', time_b - time_a)

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文