kundajelab / basepairmodels

MIT License
16 stars 6 forks source link

using tensorflow 2 #25

Closed nzhang89 closed 2 years ago

nzhang89 commented 2 years ago

Thank you for developing basepairmodels. We are interested in using it on some of our ChIP-seq and CUT&Tag data.

I have a question about tensorflow version. We recently built a GPU server with A100 GPU cards. The minimum CUDA version for A100 is 11. If I install the package using pip install git+https://github.com/kundajelab/basepairmodels.git@v0.1.4 based on the README file, I get tensorflow 1.14 (it works on CUDA10 and A100, but not optimal).

Is there a way to use tensorflow 2 with CUDA11? For example, if I directly install the dev version with pip install git+https://github.com/kundajelab/basepairmodels.git (remove the @v0.1.4), the tensorflow is bumped to 2.4.1 which can be paired with CUDA11. I am wondering if this works or not.

Thank you so much for your help. Let me know if you have any questions.

Ning