前提・実現したいこと
Sketch Your Own GANのリンク内容
を実行している際に、エラーが発生しました。
解決方法をお伺いしたいです。
python generate.py --ckpt weights/photosketch_standing_cat_noaug.pth --save_dir output/samples_standing_cat
を実行しました。
発生している問題・エラーメッセージ
/home/a/libraries/anaconda3/envs/gan/lib/python3.6/site-packages/torch/cuda/__init__.py:80: UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 10000). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver. (Triggered internally at ../c10/cuda/CUDAFunctions.cpp:112.) return torch._C._cuda_getDeviceCount() > 0 No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda' Traceback (most recent call last): File "generate.py", line 8, in <module> from training.networks.stylegan2 import Generator File "/home/a/GANSketching/training/networks/__init__.py", line 2, in <module> from .misc import * File "/home/a/GANSketching/training/networks/misc.py", line 1, in <module> from . import stylegan2 File "/home/a/GANSketching/training/networks/stylegan2.py", line 8, in <module> from .op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d File "/home/a/GANSketching/training/networks/op/__init__.py", line 1, in <module> from .fused_act import FusedLeakyReLU, fused_leaky_relu File "/home/a/GANSketching/training/networks/op/fused_act.py", line 15, in <module> os.path.join(module_path, "fused_bias_act_kernel.cu"), File "/home/a/libraries/anaconda3/envs/gan/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1136, in load keep_intermediates=keep_intermediates) File "/home/a/libraries/anaconda3/envs/gan/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1347, in _jit_compile is_standalone=is_standalone) File "/home/a/libraries/anaconda3/envs/gan/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1445, in _write_ninja_file_and_build_library is_standalone=is_standalone) File "/home/a/libraries/anaconda3/envs/gan/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1834, in _write_ninja_file_to_build_library cuda_flags = common_cflags + COMMON_NVCC_FLAGS + _get_cuda_arch_flags() File "/home/a/libraries/anaconda3/envs/gan/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1606, in _get_cuda_arch_flags arch_list[-1] += '+PTX' IndexError: list index out of range