search

ANTsX / ANTsPyNet

Pre-trained models and utilities for deep learning on medical images in Python
https://antspynet.readthedocs.io
Apache License 2.0
201 stars 29 forks source link

Requirements for pypi releases #110

Closed cookpa closed 2 months ago

cookpa commented 6 months ago 
          Fantastic! This will help me a ton. Need help getting a pypi release up? Here is the [GHA workflow](https://github.com/nitrain/nitrain/blob/main/.github/workflows/python-publish.yml) I use personally to automatically push to pypi every time you make a new release on github. Just requires setting the pypi token as a repo secret.

Originally posted by @ncullen93 in https://github.com/ANTsX/ANTsPyNet/issues/93#issuecomment-2080379979

cookpa commented 6 months ago

@ncullen93 thanks for this - starting a new issue for it

cookpa commented 6 months ago

One thing I wanted to ask (re: #108) is whether the requirements should be tensorflow-gpu instead of tensorflow.

I'm less familiar with keras, is there an equivalent like keras-gpu? Or does normal keras just work?

ntustison commented 6 months ago

GPU vs. no-GPU is a tensorflow distinction. This used to be a big headache when I would try to coordinate packages in ANTsRNet as it sits on a Python layer. However, it seems as though that has been recently much better sorted out in an automatic fashion.

cookpa commented 6 months ago

Huh, looks like tensorflow-gpu is now deprecated. So hopefully we're good on that front. I guess it's down to individual users to make sure they install a tensorflow that is compatible with their cuda version according to

https://www.tensorflow.org/install/source#gpu

ntustison commented 6 months ago

Good to know. Thanks.