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

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

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

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

主要代码如下:

# -*- 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)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值