tbepler / topaz

Pipeline for particle picking in cryo-electron microscopy images using convolutional neural networks trained from positive and unlabeled examples. Also featuring micrograph and tomogram denoising with DNNs.
GNU General Public License v3.0
172 stars 64 forks source link

Support for Cuda 11 #98

Closed alexjnoble closed 3 years ago

alexjnoble commented 3 years ago

Oli reports here that Cuda 11.2 seems to not be compatible with Topaz:

https://discuss.cryosparc.com/t/topaz-train-error-in-csparc-3-2/6500

Can support be added?

Thanks, -Alex

tbepler commented 3 years ago

Looks like the cryoSPARC team got topaz to run with CUDA 11.1, so I don't think this is a topaz issue. Looks like it was just an installation problem with pytorch.