相关疑难解决方法(0)

我在 Tensorflow 中设置了一个简单的线性回归问题，并在 1.13.1 中使用 Tensorflow CPU 和 GPU 创建了简单的 conda 环境（在 NVIDIA Quadro P600 的后端使用 CUDA 10.0）。

但是，看起来 GPU 环境总是比 CPU 环境花费更长的时间。我正在运行的代码如下。

import time
import warnings
import numpy as np
import scipy

import tensorflow as tf
import tensorflow_probability as tfp

from tensorflow_probability import edward2 as ed
from tensorflow.python.ops import control_flow_ops
from tensorflow_probability import distributions as tfd



# Handy snippet to reset the global graph and global session.
def reset_g():
    with warnings.catch_warnings():
        warnings.simplefilter('ignore')
        tf.reset_default_graph()
        try:
            sess.close()
        except:
            pass


N = …