Traceback (most recent call last):
File “train_stylegan2.py”, line 402, in
worker(P)
File “train_stylegan2.py”, line 311, in worker
generator, discriminator = get_architecture(P.architecture, image_size, P=P)
File “/teamspace/studios/this_studio/ViTGAN/models/gan/init.py”, line 29, in get_architecture
from models.gan.stylegan2.vit_generator import Generator
File “/teamspace/studios/this_studio/ViTGAN/models/gan/stylegan2/vit_generator.py”, line 12, in
from models.gan.stylegan2.op import FusedLeakyReLU, conv2d_gradfix
File “/teamspace/studios/this_studio/ViTGAN/models/gan/stylegan2/op/init.py”, line 1, in
from .fused_act import FusedLeakyReLU, fused_leaky_relu
File “/teamspace/studios/this_studio/ViTGAN/models/gan/stylegan2/op/fused_act.py”, line 15, in
os.path.join(module_path, “fused_bias_act_kernel.cu”),
File “/home/zeus/miniconda3/envs/cloudspace/lib/python3.7/site-packages/torch/utils/cpp_extension.py”, line 1296, in load
keep_intermediates=keep_intermediates)
File “/home/zeus/miniconda3/envs/cloudspace/lib/python3.7/site-packages/torch/utils/cpp_extension.py”, line 1518, in _jit_compile
is_standalone=is_standalone)
File “/home/zeus/miniconda3/envs/cloudspace/lib/python3.7/site-packages/torch/utils/cpp_extension.py”, line 1619, in _write_ninja_file_and_build_library
is_standalone=is_standalone)
File “/home/zeus/miniconda3/envs/cloudspace/lib/python3.7/site-packages/torch/utils/cpp_extension.py”, line 2014, in _write_ninja_file_to_build_library
cuda_flags = common_cflags + COMMON_NVCC_FLAGS + _get_cuda_arch_flags()
File “/home/zeus/miniconda3/envs/cloudspace/lib/python3.7/site-packages/torch/utils/cpp_extension.py”, line 1793, in _get_cuda_arch_flags
raise ValueError(f"Unknown CUDA arch ({arch}) or GPU not supported")
ValueError: Unknown CUDA arch (9.0) or GPU not supported