search

skoblov-lab / spliceai-reforged

A GPU-friendly implementation of SpliceAI
10 stars 2 forks source link

Newer GPUs #5

Open drmurdock opened 9 months ago

drmurdock commented 9 months ago

Hi, thank you for putting spliceai_reforged together. Have you had any experience running it with more modern GPUs? I have a GTX 3060 Ti and have been unable to get it to work despite everything being installed correctly. I'm getting the same problem referenced below (where I also posted) using the original Illumina implementation. It may be related to using newer GPUs but I don't have an older one to confirm. Thanks.

https://github.com/Illumina/SpliceAI/issues/99

edikedik commented 9 months ago

Hi, @drmurdock No, we haven't tried that. As the code here is a wrapper around the original SpliceAI models, one would expect to obtain the same result on the same input. Unfortunately, this implies dragging along hardware-related issues such as the one you referenced. I'm not sure we could offer a solution here. It's likely that the issue is related to an interplay between newer tensorflow and the GPU driver. For the record, the models were written with tensorflow=1.2.0 and were still functional on v2.0, and the latest GPU we tried it with was the "GeForce Titan X" card.

drmurdock commented 9 months ago

Thanks @edikedik. I will try to swap out my card for an older, compatible one. Looking forward to running your tool!