简体   繁体   中英

Tensorflow GPU cuDNn: How do I load cuDDN libraries?

I am trying to use tensorflow with gpu and installed CUDA 8.0 toolkit and cuDNn v5.1 libraries as described in nvidia website. But when I try to import tensorflow as module in python3.5, it does not load cuDNn libraries (outputs nothing, just loads tensorflow module). And I do not observe speed in processing (same speed I obtained when I use CPU) with GPU.

Fresh install is the key but there are some important points: 1. Install CUDA 8.0 toolkit 2. Install cuDNn 5.1 version (not 6.0) 3. Install from source(bazel) and configure to use ensorflow with CUDA

Above steps worked for me! Hope it helps anyone.

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM