NVlabs:stylegan2による学習する際に発生するエラー

前提・実現したいこと

NVlabs(https://github.com/NVlabs/stylegan2.git)を活用して、
stylegan2の学習モデルの作成〜作成したモデルを活用した画像の生成までを行えればと考えております。

NVlabs(https://github.com/NVlabs/stylegan2.git)内の、
run_training.pyを活用して、学習を行なったところ、
「self._traceback = tf_stack.extract_stack()」のようなエラーが発生しました。

発生している問題・エラーメッセージ

エラーメッセージ
Traceback (most recent call last):
File "run_training.py", line 192, in <module>
main()
File "run_training.py", line 187, in main
run(**vars(args))
File "run_training.py", line 120, in run
dnnlib.submit_run(**kwargs)
File "/content/stylegan2/dnnlib/submission/submit.py", line 343, in submit_run
return farm.submit(submit_config, host_run_dir)
File "/content/stylegan2/dnnlib/submission/internal/local.py", line 22, in submit
return run_wrapper(submit_config)
File "/content/stylegan2/dnnlib/submission/submit.py", line 280, in run_wrapper
run_func_obj(**submit_config.run_func_kwargs)
File "/content/stylegan2/training/training_loop.py", line 197, in training_loop
G_gpu = G if gpu == 0 else G.clone(G.name + '_shadow')
File "/content/stylegan2/dnnlib/tflib/network.py", line 312, in clone
net.init_graph()
File "/content/stylegan2/dnnlib/tflib/network.py", line 154, in init_graph
out_expr = self.build_func(*self.input_templates, **build_kwargs)
File "/content/stylegan2/training/networks_stylegan2.py", line 186, in G_main
components.synthesis = tflib.Network('G_synthesis', func_name=globals()[synthesis_func], **kwargs)
File "/content/stylegan2/dnnlib/tflib/network.py", line 98, in init
self.reset_own_vars()
File "/content/stylegan2/dnnlib/tflib/network.py", line 189, in reset_own_vars
tfutil.run([var.initializer for var in self.own_vars.values()])
File "/content/stylegan2/dnnlib/tflib/tfutil.py", line 31, in run
return tf.get_default_session().run(*args, **kwargs)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/client/session.py", line 956, in run
run_metadata_ptr)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/client/session.py", line 1180, in run
feed_dict_tensor, options, run_metadata)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/client/session.py", line 1359, in do_run
run_metadata)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/client/session.py", line 1384, in do_call
raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot assign a device for operation G_synthesis_2/noise0/Initializer/random_normal/RandomStandardNormal: Could not satisfy explicit device specification '' because the node node G_synthesis_2/noise0/Initializer/random_normal/RandomStandardNormal (defined at /tensorflow-1.15.2/python3.7/tensorflow_core/python/framework/ops.py:1748) placed on device Device assignments active during op 'G_synthesis_2/noise0/Initializer/random_normal/RandomStandardNormal' creation:
with tf.device(None): </tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variables.py:1816>
with tf.device(/gpu:1): </content/stylegan2/training/training_loop.py:194> was colocated with a group of nodes that required incompatible device '/device:GPU:1'. All available devices [/job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:XLA_CPU:0, /job:localhost/replica:0/task:0/device:XLA_GPU:0, /job:localhost/replica:0/task:0/device:GPU:0].
Colocation Debug Info:
Colocation group had the following types and supported devices:
Root Member(assigned_device_name_index=-1 requested_device_name='/device:GPU:1' assigned_device_name='' resource_device_name='/device:GPU:1' supported_device_types=[GPU, CPU] possible_devices=[]
Assign: GPU CPU
Mul: GPU CPU XLA_CPU XLA_GPU
Const: GPU CPU XLA_CPU XLA_GPU
VariableV2: GPU CPU
Add: GPU CPU XLA_CPU XLA_GPU
RandomStandardNormal: GPU CPU XLA_CPU XLA_GPU

Original stack trace for 'G_synthesis_2/noise0/Initializer/random_normal/RandomStandardNormal':
File "run_training.py", line 192, in <module>
main()
File "run_training.py", line 187, in main
run(**vars(args))
File "run_training.py", line 120, in run
dnnlib.submit_run(**kwargs)
File "/content/stylegan2/dnnlib/submission/submit.py", line 343, in submit_run
return farm.submit(submit_config, host_run_dir)
File "/content/stylegan2/dnnlib/submission/internal/local.py", line 22, in submit
return run_wrapper(submit_config)
File "/content/stylegan2/dnnlib/submission/submit.py", line 280, in run_wrapper
run_func_obj(**submit_config.run_func_kwargs)
File "/content/stylegan2/training/training_loop.py", line 197, in training_loop
G_gpu = G if gpu == 0 else G.clone(G.name + '_shadow')
File "/content/stylegan2/dnnlib/tflib/network.py", line 312, in clone
net._init_graph()
File "/content/stylegan2/dnnlib/tflib/network.py", line 154, in _init_graph
out_expr = self._build_func(*self.input_templates, **build_kwargs)
File "/content/stylegan2/training/networks_stylegan2.py", line 186, in G_main
components.synthesis = tflib.Network('G_synthesis', func_name=globals()[synthesis_func], **kwargs)
File "/content/stylegan2/dnnlib/tflib/network.py", line 97, in init
self._init_graph()
File "/content/stylegan2/dnnlib/tflib/network.py", line 154, in _init_graph
out_expr = self._build_func(*self.input_templates, **build_kwargs)
File "/content/stylegan2/training/networks_stylegan2.py", line 451, in G_synthesis_stylegan2
noise_inputs.append(tf.get_variable('noise%d' % layer_idx, shape=shape, initializer=tf.initializers.random_normal(), trainable=False))
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variable_scope.py", line 1500, in get_variable
aggregation=aggregation)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variable_scope.py", line 1243, in get_variable
aggregation=aggregation)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variable_scope.py", line 567, in get_variable
aggregation=aggregation)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variable_scope.py", line 519, in _true_getter
aggregation=aggregation)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variable_scope.py", line 933, in _get_single_variable
aggregation=aggregation)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variables.py", line 258, in call
return cls._variable_v1_call(*args, **kwargs)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variables.py", line 219, in _variable_v1_call
shape=shape)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variables.py", line 197, in <lambda>
previous_getter = lambda **kwargs: default_variable_creator(None, **kwargs)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variable_scope.py", line 2519, in default_variable_creator
shape=shape)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variables.py", line 262, in call
return super(VariableMetaclass, cls).call(*args, **kwargs)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variables.py", line 1688, in init
shape=shape)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variables.py", line 1818, in _init_from_args
initial_value(), name="initial_value", dtype=dtype)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/variable_scope.py", line 905, in <lambda>
partition_info=partition_info)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/init_ops.py", line 323, in call
shape, self.mean, self.stddev, dtype, seed=self.seed)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/random_ops.py", line 74, in random_normal
shape_tensor, dtype, seed=seed1, seed2=seed2)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/ops/gen_random_ops.py", line 729, in random_standard_normal
seed2=seed2, name=name)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/framework/op_def_library.py", line 794, in _apply_op_helper
op_def=op_def)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/util/deprecation.py", line 507, in new_func
return func(*args, **kwargs)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/framework/ops.py", line 3357, in create_op
attrs, op_def, compute_device)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/framework/ops.py", line 3426, in _create_op_internal
op_def=op_def)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/framework/ops.py", line 1748, in init
self._traceback = tf_stack.extract_stack()