No CUDA runtime is found, using CUDA_HOME='/cm/shared/apps/cuda100/10.0.130'
Traceback (most recent call last):
File "applications/propagate_to_images.py", line 20, in <module>
from applications import base_eval_argparse, load_stn, determine_flips
File "/home/user/nvidia_code/gangealing/applications/__init__.py", line 3, in <module>
from models import get_stn, ResnetClassifier
File "/home/user/nvidia_code/gangealing/models/__init__.py", line 3, in <module>
from models.spatial_transformers.spatial_transformer import get_stn, ComposedSTN, SpatialTransformer
File "/home/user/nvidia_code/gangealing/models/spatial_transformers/spatial_transformer.py", line 5, in <module>
from models.stylegan2.networks import EqualLinear, ConvLayer, ResBlock
File "/home/user/nvidia_code/gangealing/models/stylegan2/networks.py", line 6, in <module>
from models.stylegan2.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d, conv2d_gradfix
File "/home/user/nvidia_code/gangealing/models/stylegan2/op/__init__.py", line 1, in <module>
from .fused_act import FusedLeakyReLU, fused_leaky_relu
File "/home/user/nvidia_code/gangealing/models/stylegan2/op/fused_act.py", line 11, in <module>
fused = load(
File "/home/user/anaconda2/envs/gg/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1144, in load
return _jit_compile(
File "/home/user/anaconda2/envs/gg/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1357, in _jit_compile
_write_ninja_file_and_build_library(
File "/home/user/anaconda2/envs/gg/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1456, in _write_ninja_file_and_build_library
_write_ninja_file_to_build_library(
File "/home/user/anaconda2/envs/gg/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1857, in _write_ninja_file_to_build_library
cuda_flags = common_cflags + COMMON_NVCC_FLAGS + _get_cuda_arch_flags()
File "/home/user/anaconda2/envs/gg/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1626, in _get_cuda_arch_flags
arch_list[-1] += '+PTX'
THe command I run is : python applications/propagate_to_images.py --ckpt cat --real_data_path data/lsun_cats --real_size 512 --dset_indices 1922 2363 8558 7401 9750 7432 2105 53 1946
Any help would be appreciated.