CSBDeep / CSBDeep_fiji

BSD 2-Clause "Simplified" License
11 stars 4 forks source link

GPU support on Windows #24

Closed frauzufall closed 5 years ago

frauzufall commented 5 years ago

We need to add installation instructions for Windows GPU support (and also update the ones for Linux).

Until we finish the PR to TensorFlow automatically loading the GPU or CPU lib depending on what is supported, the workaround is to manually add the native GPU TensorFlow library to Fiji.

Here is an update site that should do this for Windows:

https://sites.imagej.net/CSBDeep-GPU-win64-TF1.12.0

In Linux I also have to add CUDA to my PATH and LD_LIBRARY_PATH environment variables, something similar might have to be done for Windows:

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} 

Could someone on Windows with a recent CUDA installation add this update site additionally to the default CSBDeep update site and try if it works or what else needs to be done (e.g. environment variables)?

I would update the documentation then asap.

maweigert commented 5 years ago

Just tested it on Windows 7, Cuda 9.0, Cudnn 7.3 and after activating the update site everything runs on the GPU (without any path adjustments), woohoo! Well done!

frauzufall commented 5 years ago

Cool, thanks for testing!

I also updated the Wiki install page at least a bit to reflect that there is also GPU support for Windows and removed the old JNI CUDA update sites.

Would be great to also test Windows 10.

WillGiang commented 5 years ago

Hi, the CSBDeep Fiji plugin works on Win10, CUDA 9.0, CuDNN 7.4.2, with the update site.

lacan commented 5 years ago

Hi all,

I'm running CARE using CUDA 10.0 and CUDNN 7.6.0 and getting tensorflow_jni.dll from https://www.tensorflow.org/install/lang_java

Works like a charm!

Oli

FengZhiheng commented 5 years ago

Installing TensorFlow GPU native libraries on Windows. I meet some troubles: image I really need your help.

FengZhiheng commented 5 years ago

I have solved this problem by changing "https" to "http", it works "http://sites.imagej.net/CSBDeep-GPU-win64-TF1.12.0"

FengZhiheng commented 5 years ago

Sorry, I have new problem: I have already install "tensorflow_jni.dll" for fiji. But tensorflow still doesn't work. I'm so confused. image

fjug commented 5 years ago

@frauzufall Help?!