HF Space: RuntimeError: CUDA error: no kernel image is available for execution on the device

Spaces
1

My requirements, as per the docs, contain the following lines:

--extra-index-url https://download.pytorch.org/whl/cu113
torch

However, my Space cannot build with A10G GPUs because of a CUDA kernel error:

Traceback (most recent call last):
  File "app.py", line 88, in <module>
    cg = ClimateGAN(
  File "/home/user/app/climategan_wrapper.py", line 143, in __init__
    self.trainer.G.half()
  File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/torch/nn/modules/module.py", line 506, in half
    return self._apply(lambda t: t.half() if t.is_floating_point() else t)
  File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/torch/nn/modules/module.py", line 359, in _apply
    module._apply(fn)
  File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/torch/nn/modules/module.py", line 359, in _apply
    module._apply(fn)
  File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/torch/nn/modules/module.py", line 381, in _apply
    param_applied = fn(param)
  File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/torch/nn/modules/module.py", line 506, in <lambda>
    return self._apply(lambda t: t.half() if t.is_floating_point() else t)
RuntimeError: CUDA error: no kernel image is available for execution on the device