search

skyflynil / stylegan2

StyleGAN2 - Official TensorFlow Implementation with practical improvements
http://arxiv.org/abs/1912.04958
Other
120 stars 33 forks source link

tensorflow.python.framework.errors_impl.NotFoundError #28

Open futureflsl opened 3 years ago

futureflsl commented 3 years ago

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?