barisgecer / OSTeC

TF implementation of our CVPR 2021 paper: OSTeC: One-Shot Texture Completion
https://openaccess.thecvf.com/content/CVPR2021/html/Gecer_OSTeC_One-Shot_Texture_Completion_CVPR_2021_paper.html
Other
192 stars 28 forks source link

Crashing while running the program: same configuration that this repository was tested with #21

Open manifischer opened 2 years ago

manifischer commented 2 years ago

Hi, I followed your installation and environment requirements. I pass all the model loading and compilations successfully, but then in line 95 in file arcface_handler.py: return self.sess.run(self.emb_norm, {self.img_ph: [img.pixels_with_channels_at_back()], self.lms_ph:[lms]}) it crashes with error: Check failed: cusolverDnCreate(&cusolver_dn_handle) == CUSOLVER_STATUS_SUCCESS Failed to create cuSolverDN instance any idea why this happens? (tried to google it, but did not find anything that was not in the code already as config.gpu_options.allow_growth = True Any idea how I can fix this? Thanks

Storm0921 commented 1 year ago

I have the same problem? How is it solved? Thanks