[时间换显存]DeepFaceLabRG.2022_2022.7.27,小显存卡跑起来

sky10080 · 发表于 2022-4-25 23:26:25

感谢分享

jinjin327 · 发表于 2022-4-28 19:27:05

这个是真的好东西

jinjin327 · 发表于 2022-4-29 16:59:56

请问这是啥原因？

Error: 2 root error(s) found.
  (0) Resource exhausted: OOM when allocating tensor with shape[736,100,100] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
      [[node Pad_6 (defined at \DeepFaceLabRG\core\leras\layers\Conv2D.py:87) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

      [[concat_1/concat/_939]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

  (1) Resource exhausted: OOM when allocating tensor with shape[736,100,100] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
      [[node Pad_6 (defined at \DeepFaceLabRG\core\leras\layers\Conv2D.py:87) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

0 successful operations.
0 derived errors ignored.

Errors may have originated from an input operation.
Input Source operations connected to node Pad_6:
LeakyRelu_5 (defined at \DeepFaceLabRG\core\leras\archis\DeepFakeArchi.py:29)

Input Source operations connected to node Pad_6:
LeakyRelu_5 (defined at \DeepFaceLabRG\core\leras\archis\DeepFakeArchi.py:29)

Original stack trace for 'Pad_6':
  File "\python-3.8.5\lib\threading.py", line 890, in _bootstrap
self._bootstrap_inner()
  File "\python-3.8.5\lib\threading.py", line 932, in _bootstrap_inner
self.run()
  File "\python-3.8.5\lib\threading.py", line 870, in run
self._target(*self._args, **self._kwargs)
  File "\DeepFaceLabRG\mainscripts\Trainer.py", line 45, in trainerThread
model = models.import_model(model_class_name)(
  File "\DeepFaceLabRG\models\ModelBase.py", line 195, in __init__
self.on_initialize()
  File "\DeepFaceLabRG\models\Model_SAEHD\Model.py", line 408, in on_initialize
gpu_dst_code    = self.inter(self.encoder(gpu_warped_dst))
  File "\DeepFaceLabRG\core\leras\models\ModelBase.py", line 117, in __call__
return self.forward(*args, **kwargs)
  File "\DeepFaceLabRG\core\leras\ops\__init__.py", line 33, in inner
return call_and_grad(*inputs)
  File "\python-3.8.5\lib\site-packages\tensorflow\python\ops\custom_gradient.py", line 261, in __call__
return self._d(self._f, a, k)
  File "\python-3.8.5\lib\site-packages\tensorflow\python\ops\custom_gradient.py", line 217, in decorated
return _graph_mode_decorator(wrapped, args, kwargs)
  File "\python-3.8.5\lib\site-packages\tensorflow\python\ops\custom_gradient.py", line 330, in _graph_mode_decorator
result, grad_fn = f(*args)
  File "\DeepFaceLabRG\core\leras\ops\__init__.py", line 20, in call_and_grad
outputs = kernel_call()
  File "\DeepFaceLabRG\core\leras\ops\__init__.py", line 19, in kernel_call
return call(self, *inputs, **kwargs)
  File "\DeepFaceLabRG\core\leras\archis\DeepFakeArchi.py", line 119, in forward
x = self.down1(x)
  File "\DeepFaceLabRG\core\leras\models\ModelBase.py", line 117, in __call__
return self.forward(*args, **kwargs)
  File "\DeepFaceLabRG\core\leras\archis\DeepFakeArchi.py", line 63, in forward
x = down(x)
  File "\DeepFaceLabRG\core\leras\models\ModelBase.py", line 117, in __call__
return self.forward(*args, **kwargs)
  File "\DeepFaceLabRG\core\leras\archis\DeepFakeArchi.py", line 43, in forward
x = self.conv1(x)
  File "\DeepFaceLabRG\core\leras\layers\LayerBase.py", line 14, in __call__
return self.forward(*args, **kwargs)
  File "\DeepFaceLabRG\core\leras\layers\Conv2D.py", line 87, in forward
x = tf.pad (x, padding, mode='CONSTANT')
  File "\python-3.8.5\lib\site-packages\tensorflow\python\util\dispatch.py", line 206, in wrapper
return target(*args, **kwargs)
  File "\python-3.8.5\lib\site-packages\tensorflow\python\ops\array_ops.py", line 3514, in pad
result = gen_array_ops.pad(tensor, paddings, name=name)
  File "\python-3.8.5\lib\site-packages\tensorflow\python\ops\gen_array_ops.py", line 6503, in pad
_, _, _op, _outputs = _op_def_library._apply_op_helper(
  File "\python-3.8.5\lib\site-packages\tensorflow\python\framework\op_def_library.py", line 748, in _apply_op_helper
op = g._create_op_internal(op_type_name, inputs, dtypes=None,
  File "\python-3.8.5\lib\site-packages\tensorflow\python\framework\ops.py", line 3557, in _create_op_internal
ret = Operation(
  File "\python-3.8.5\lib\site-packages\tensorflow\python\framework\ops.py", line 2045, in __init__
self._traceback = tf_stack.extract_stack_for_node(self._c_op)

Traceback (most recent call last):
  File "D:\DeepFaceLab\_internal\python-3.8.5\lib\site-packages\tensorflow\python\client\session.py", line 1375, in _do_call
return fn(*args)
  File "D:\DeepFaceLab\_internal\python-3.8.5\lib\site-packages\tensorflow\python\client\session.py", line 1359, in _run_fn
return self._call_tf_sessionrun(options, feed_dict, fetch_list,
  File "D:\DeepFaceLab\_internal\python-3.8.5\lib\site-packages\tensorflow\python\client\session.py", line 1451, in _call_tf_sessionrun
return tf_session.TF_SessionRun_wrapper(self._session, options, feed_dict,
tensorflow.python.framework.errors_impl.ResourceExhaustedError: 2 root error(s) found.
  (0) Resource exhausted: OOM when allocating tensor with shape[736,100,100] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
      [[{{node Pad_6}}]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

      [[concat_1/concat/_939]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

  (1) Resource exhausted: OOM when allocating tensor with shape[736,100,100] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
      [[{{node Pad_6}}]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

0 successful operations.
0 derived errors ignored.

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "D:\DeepFaceLab\_internal\DeepFaceLabRG\mainscripts\Trainer.py", line 130, in trainerThread
iter, iter_time = model.train_one_iter()
  File "D:\DeepFaceLab\_internal\DeepFaceLabRG\models\ModelBase.py", line 476, in train_one_iter
losses = self.onTrainOneIter()
  File "D:\DeepFaceLab\_internal\DeepFaceLabRG\models\Model_SAEHD\Model.py", line 774, in onTrainOneIter
src_loss, dst_loss = self.src_dst_train (warped_src, target_src, target_srcm, target_srcm_em, warped_dst, target_dst, target_dstm, target_dstm_em)
  File "D:\DeepFaceLab\_internal\DeepFaceLabRG\models\Model_SAEHD\Model.py", line 576, in src_dst_train
s, d = nn.tf_sess.run ( [ src_loss, dst_loss, src_dst_loss_gv_op],
  File "D:\DeepFaceLab\_internal\python-3.8.5\lib\site-packages\tensorflow\python\client\session.py", line 967, in run
result = self._run(None, fetches, feed_dict, options_ptr,
  File "D:\DeepFaceLab\_internal\python-3.8.5\lib\site-packages\tensorflow\python\client\session.py", line 1190, in _run
results = self._do_run(handle, final_targets, final_fetches,
  File "D:\DeepFaceLab\_internal\python-3.8.5\lib\site-packages\tensorflow\python\client\session.py", line 1368, in _do_run
return self._do_call(_run_fn, feeds, fetches, targets, options,
  File "D:\DeepFaceLab\_internal\python-3.8.5\lib\site-packages\tensorflow\python\client\session.py", line 1394, in _do_call
raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.ResourceExhaustedError: 2 root error(s) found.
  (0) Resource exhausted: OOM when allocating tensor with shape[736,100,100] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
      [[node Pad_6 (defined at \DeepFaceLabRG\core\leras\layers\Conv2D.py:87) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

      [[concat_1/concat/_939]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

  (1) Resource exhausted: OOM when allocating tensor with shape[736,100,100] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
      [[node Pad_6 (defined at \DeepFaceLabRG\core\leras\layers\Conv2D.py:87) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

0 successful operations.
0 derived errors ignored.

Errors may have originated from an input operation.
Input Source operations connected to node Pad_6:
LeakyRelu_5 (defined at \DeepFaceLabRG\core\leras\archis\DeepFakeArchi.py:29)

Input Source operations connected to node Pad_6:
LeakyRelu_5 (defined at \DeepFaceLabRG\core\leras\archis\DeepFakeArchi.py:29)

Original stack trace for 'Pad_6':
  File "\python-3.8.5\lib\threading.py", line 890, in _bootstrap
self._bootstrap_inner()
  File "\python-3.8.5\lib\threading.py", line 932, in _bootstrap_inner
self.run()
  File "\python-3.8.5\lib\threading.py", line 870, in run
self._target(*self._args, **self._kwargs)
  File "\DeepFaceLabRG\mainscripts\Trainer.py", line 45, in trainerThread
model = models.import_model(model_class_name)(
  File "\DeepFaceLabRG\models\ModelBase.py", line 195, in __init__
self.on_initialize()
  File "\DeepFaceLabRG\models\Model_SAEHD\Model.py", line 408, in on_initialize
gpu_dst_code    = self.inter(self.encoder(gpu_warped_dst))
  File "\DeepFaceLabRG\core\leras\models\ModelBase.py", line 117, in __call__
return self.forward(*args, **kwargs)
  File "\DeepFaceLabRG\core\leras\ops\__init__.py", line 33, in inner
return call_and_grad(*inputs)
  File "\python-3.8.5\lib\site-packages\tensorflow\python\ops\custom_gradient.py", line 261, in __call__
return self._d(self._f, a, k)
  File "\python-3.8.5\lib\site-packages\tensorflow\python\ops\custom_gradient.py", line 217, in decorated
return _graph_mode_decorator(wrapped, args, kwargs)
  File "\python-3.8.5\lib\site-packages\tensorflow\python\ops\custom_gradient.py", line 330, in _graph_mode_decorator
result, grad_fn = f(*args)
  File "\DeepFaceLabRG\core\leras\ops\__init__.py", line 20, in call_and_grad
outputs = kernel_call()
  File "\DeepFaceLabRG\core\leras\ops\__init__.py", line 19, in kernel_call
return call(self, *inputs, **kwargs)
  File "\DeepFaceLabRG\core\leras\archis\DeepFakeArchi.py", line 119, in forward
x = self.down1(x)
  File "\DeepFaceLabRG\core\leras\models\ModelBase.py", line 117, in __call__
return self.forward(*args, **kwargs)
  File "\DeepFaceLabRG\core\leras\archis\DeepFakeArchi.py", line 63, in forward
x = down(x)
  File "\DeepFaceLabRG\core\leras\models\ModelBase.py", line 117, in __call__
return self.forward(*args, **kwargs)
  File "\DeepFaceLabRG\core\leras\archis\DeepFakeArchi.py", line 43, in forward
x = self.conv1(x)
  File "\DeepFaceLabRG\core\leras\layers\LayerBase.py", line 14, in __call__
return self.forward(*args, **kwargs)
  File "\DeepFaceLabRG\core\leras\layers\Conv2D.py", line 87, in forward
x = tf.pad (x, padding, mode='CONSTANT')
  File "\python-3.8.5\lib\site-packages\tensorflow\python\util\dispatch.py", line 206, in wrapper
return target(*args, **kwargs)
  File "\python-3.8.5\lib\site-packages\tensorflow\python\ops\array_ops.py", line 3514, in pad
result = gen_array_ops.pad(tensor, paddings, name=name)
  File "\python-3.8.5\lib\site-packages\tensorflow\python\ops\gen_array_ops.py", line 6503, in pad
_, _, _op, _outputs = _op_def_library._apply_op_helper(
  File "\python-3.8.5\lib\site-packages\tensorflow\python\framework\op_def_library.py", line 748, in _apply_op_helper
op = g._create_op_internal(op_type_name, inputs, dtypes=None,
  File "\python-3.8.5\lib\site-packages\tensorflow\python\framework\ops.py", line 3557, in _create_op_internal
ret = Operation(
  File "\python-3.8.5\lib\site-packages\tensorflow\python\framework\ops.py", line 2045, in __init__
self._traceback = tf_stack.extract_stack_for_node(self._c_op)