search

dailenson / SDT

This repository is the official implementation of Disentangling Writer and Character Styles for Handwriting Generation (CVPR 2023)
MIT License
968 stars 82 forks source link

cuDNN error #25

Closed Barthold-Albrecht closed 11 months ago

Barthold-Albrecht commented 11 months ago

Running inference with the pre-trained model on a VM (Tesla K80) gives RuntimeError: cuDNN error: CUDNN_STATUS_ARCH_MISMATCH. I set up the env with your requirements. Do you have any hints on what might go wrong here?

dailenson commented 11 months ago

Could you please provide your complete pytorch environment, like cuda, cuDNN, pytorch version? I guess your incorrect cuDNN version causes this error.

Barthold-Albrecht commented 11 months ago

Thx for getting back. Issue is solved. You were right, setting cuda=11.8 did the trick.