-
Notifications
You must be signed in to change notification settings - Fork 33
Description
when I run:
python run_generator.py generate-images --network=weights/stylegan2-car-config-f.pkl --seeds=6000-6025 --truncation-psi=0.5
error occurd:
Traceback (most recent call last):
File "run_generator.py", line 168, in
main()
File "run_generator.py", line 163, in main
dnnlib.submit_run(sc, func_name_map[subcmd], **kwargs)
File "/home/fut/Downloads/stylegan2-NVlabs/dnnlib/submission/submit.py", line 343, in submit_run
return farm.submit(submit_config, host_run_dir)
File "/home/fut/Downloads/stylegan2-NVlabs/dnnlib/submission/internal/local.py", line 22, in submit
return run_wrapper(submit_config)
File "/home/fut/Downloads/stylegan2-NVlabs/dnnlib/submission/submit.py", line 280, in run_wrapper
run_func_obj(**submit_config.run_func_kwargs)
File "/home/fut/Downloads/stylegan2-NVlabs/run_generator.py", line 21, in generate_images
_G, _D, Gs = pretrained_networks.load_networks(network_pkl)
File "/home/fut/Downloads/stylegan2-NVlabs/pretrained_networks.py", line 76, in load_networks
G, D, Gs = pickle.load(stream, encoding='latin1')
File "/home/fut/Downloads/stylegan2-NVlabs/dnnlib/tflib/network.py", line 297, in setstate
self._init_graph()
File "/home/fut/Downloads/stylegan2-NVlabs/dnnlib/tflib/network.py", line 154, in _init_graph
out_expr = self._build_func(*self.input_templates, **build_kwargs)
File "", line 491, in G_synthesis_stylegan2
File "", line 455, in layer
File "", line 99, in modulated_conv2d_layer
File "", line 68, in apply_bias_act
File "/home/fut/Downloads/stylegan2-NVlabs/dnnlib/tflib/ops/fused_bias_act.py", line 68, in fused_bias_act
return impl_dict[impl](x=x, b=b, axis=axis, act=act, alpha=alpha, gain=gain)
File "/home/fut/Downloads/stylegan2-NVlabs/dnnlib/tflib/ops/fused_bias_act.py", line 122, in _fused_bias_act_cuda
cuda_kernel = _get_plugin().fused_bias_act
File "/home/fut/Downloads/stylegan2-NVlabs/dnnlib/tflib/ops/fused_bias_act.py", line 16, in _get_plugin
return custom_ops.get_plugin(os.path.splitext(file)[0] + '.cu')
File "/home/fut/Downloads/stylegan2-NVlabs/dnnlib/tflib/custom_ops.py", line 156, in get_plugin
plugin = tf.load_op_library(bin_file)
File "/home/fut/miniconda3/envs/fut/lib/python3.7/site-packages/tensorflow/python/framework/load_library.py", line 61, in load_op_library
lib_handle = py_tf.TF_LoadLibrary(library_filename)
tensorflow.python.framework.errors_impl.NotFoundError: /home/fut/Downloads/stylegan2-NVlabs/dnnlib/tflib/_cudacache/fused_bias_act_09d1a1c1272fedad586c4a59ce52e74b.so: undefined symbol: _ZN10tensorflow12OpDefBuilder5InputESs
environment:
ubuntu18.04
tensorflow-gpu==1.14.0
I reinstall tensorflow-gpu,but the same error like before.How can solve this problem?