inference error

[qkrwjdgnl]

First I Installed the pretrained_models. And I try to run inference script

python scripts/inference.py \ --exp_dir=/path/to/experiment \ --checkpoint_path=experiment/checkpoints/best_model.pt \ --data_path=/path/to/test_data \ --test_batch_size=4 \ --test_workers=4 \ --couple_outputs --target_age=0,10,20,30,40,50,60,70,80

error is "ValueError: Unknown CUDA arch (8.6) or GPU not supported"

Does anyone know why this happens and how I can fix it?

Details like this: (SAM) sadmin@WS-E900-G4-WS980T:~/anaconda3/envs/SAM/SAM$ python scripts/inference.py \

--exp_dir=/path/to/experiment \ --checkpoint_path=experiment/checkpoints/best_model.pt \ --data_path=/path/to/test_data \ --test_batch_size=4 \ --test_workers=4 \ --couple_outputs

Traceback (most recent call last): File "scripts/inference.py", line 19, in from models.psp import pSp File "./models/psp.py", line 12, in from models.encoders import psp_encoders File "./models/encoders/psp_encoders.py", line 8, in from models.stylegan2.model import EqualLinear File "./models/stylegan2/model.py", line 7, in from models.stylegan2.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d File "./models/stylegan2/op/init.py", line 1, in from .fused_act import FusedLeakyReLU, fused_leaky_relu File "./models/stylegan2/op/fused_act.py", line 13, in os.path.join(module_path, 'fused_bias_act_kernel.cu'), File "/home/sadmin/anaconda3/envs/SAM/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 661, in load is_python_module) File "/home/sadmin/anaconda3/envs/SAM/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 830, in _jit_compile with_cuda=with_cuda) File "/home/sadmin/anaconda3/envs/SAM/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 879, in _write_ninja_file_and_build with_cuda=with_cuda) File "/home/sadmin/anaconda3/envs/SAM/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1114, in _write_ninja_file cuda_flags = common_cflags + COMMON_NVCC_FLAGS + _get_cuda_arch_flags() File "/home/sadmin/anaconda3/envs/SAM/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 991, in _get_cuda_arch_flags raise ValueError("Unknown CUDA arch ({}) or GPU not supported".format(arch)) ValueError: Unknown CUDA arch (8.6) or GPU not supported (SAM) sadmin@WS-E900-G4-WS980T:~/anaconda3/envs/SAM/SAM$ --target_age=0,10,20,30,40,50,60,70,80 --target_age=0,10,20,30,40,50,60,70,80: 명령을 찾을 수 없습니다

[qkrwjdgnl]

my cuda version is 11.2 cudnn version is 8.1.0

[qkrwjdgnl]

i solved it