Update 1.0.7 is not suitable for windows

ZFTurbo / segmentation_models_3D

Set of models for segmentation of 3D volumes

MIT License

125 stars 26 forks source link

Open rugved88 opened 10 months ago

rugved88 commented 10 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 10 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 10 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 10 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.