IsoNet-cryoET / IsoNet

Self-supervised learning for isotropic cryoET reconstruction
https://www.nature.com/articles/s41467-022-33957-8
MIT License
67 stars 12 forks source link

CUDA_ERROR_NOT_INITIALIZED #16

Open jiaxialxj opened 2 years ago

jiaxialxj commented 2 years ago

My environment is here: Ubuntu 20.04 Kernel version: 5.8.0-43-generic Nvidia Driver version: 460.27.04 CUDA Version: 11.2.0 cuDnn version: 8.1.1 Tensorflow version: 2.6.0 Screenshot from 2021-10-07 17-42-01

procyontao commented 2 years ago

Hi,

Based on our previous experience, this error could be caused by not properly loading Cudnn. Could you add "--log_level debug" in your command and check the output?

Best,