Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

CUDA® Toolkit 9.0 + cuDNN v7.0.5 #2

Open
obsydion opened this issue Jan 2, 2018 · 3 comments
Open

CUDA® Toolkit 9.0 + cuDNN v7.0.5 #2

obsydion opened this issue Jan 2, 2018 · 3 comments

Comments

@obsydion
Copy link

obsydion commented Jan 2, 2018

Hi there,
As the subject suggests I have been trying to get CSBDeep GPU acceleration to work with CUDA Toolkit 9.0 + cuDNN v7.0.5 with no avail. The reason why I didn't use Toolkit 8.0 as suggested is because I installed 9.0 to run GPU acceleration for our previously installed SVI Huygens deconvolution package.

My question is whether CSBDeep will work with Toolkit 9.0 at all? Similarly, I had to download cuDNN v7.0.5 to match Toolkit 9.0 instead of the suggested cuDNN v6.

Thanks in advance.

@tboo
Copy link
Collaborator

tboo commented Jan 3, 2018

Hi - The problem is likely not CSBdeep but Tensorflow. I believe that Tensorflow only runs on cuDNN 6.0 (unless you compile it from source). You can have both cuDNN 6.0 and 7.0 run in parallel though. I have a similar setup.

EDIT: Make sure this works: https://www.tensorflow.org/install/install_linux#run_a_short_tensorflow_program

@HedgehogCode
Copy link
Member

Hi,
As @tboo pointed out, TensorFlow requires CUDA 8 and cuDNN 6. They anticipate releasing TensorFlow 1.5 (future version) with CUDA 9 and cuDNN 7 (See the last point of the release notes of 1.4). We have to wait for this release, then update imagej-tensorflow and then we can run CSBDeep with CUDA 9 and cuDNN 7.

Also pointed out by @tboo: You can install multiple versions of CUDA and cuDNN.

@royerloic
Copy link

royerloic commented Jan 3, 2018 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants