flowdec installation downloads tensorflow when tensorflow-gpu already installed

hammerlab / flowdec

TensorFlow Deconvolution for Microscopy Data

Apache License 2.0

89 stars 26 forks source link

flowdec installation downloads tensorflow when tensorflow-gpu already installed #10

Closed elktoe closed 5 years ago

elktoe commented 5 years ago

Hi! When running the e.elegans example flowdec was running on CPU, not GPU. Also noticed that when flowdec was installed it installed tensorflow and not tensorflow-gpu.

If tensorflow was uninstalled would flowdec use tensorflow-gpu and sort itself out? Cheers

russellb commented 5 years ago

Try installing flowdec with the TF_GPU environment variable set to true.

It might be better to just document the tensorflow requirement and not try to automatically install tensorflow or tensorflow-gpu automatically.

eric-czech commented 5 years ago

For posterity here, the TF dependency was split out as an extra to be installed on request based on user preferences (see instructions in README) per fix from @russellb in https://github.com/hammerlab/flowdec/pull/11