The cuda and pytorch verision of the pytorch version CloserLook3D

[whuhxb]

Hi @zeliu98

What does the cuda and pytorch version of the pytorch version CloserLook3D work on? I have tried cuda10.1, pytorch==1.4.0 or pytorch==1.5.0 pillow=7.0 or 6.0 GPU V100, there always occur the compilation bugs like below. Thanks.

lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1316, in _get_cuda_arch_flags
raise ValueError("Unknown CUDA arch ({}) or GPU not supported".format(arch))
ValueError: Unknown CUDA arch (8.0) or GPU not supported