Update 1.0.7 is not suitable for windows

ZFTurbo / segmentation_models_3D

Set of models for segmentation of 3D volumes

MIT License

121 stars 27 forks source link

Open rugved88 opened 9 months ago

rugved88 commented 9 months ago

The requirement is 'tensorflow>=2.13.0, <=2.15.0'. However, for windows the GPU is not available for tensorflow>=2.10.0

ZFTurbo commented 9 months ago

It's available. As I remember It's included in ordinary tensor flow without _gpu suffix. I will check.

I use windows myself.

ZFTurbo commented 9 months ago

Yes, you are right...

The solution is to install previous version of package. Version 1.0.5 will do as I remember.

rugved88 commented 9 months ago

Yeah! thanks for the updates. I installed TensorFlow==2.8.0 and segmentation-models-3D==1.0.6 and it worked for me with GPU on windows.