我在华硕 K501UX 笔记本电脑上使用 NVidia Geforce GTX 950M(驱动程序 381)在 Ubuntu 16.04(x86_64)上安装了 CUDA 8.0。安装后CUDA
,我可以运行deviceQuery
and bandwidthTest
,根据this doc。
下载的 cuDNN 5.1 是一个.deb
包并自动安装。但是,没有cudnn.h
in /usr/local/cuda-8.0/include/
,也没有libcudnn
in/usr/lib/x86_64-linux-gnu/
目录。此验证表明 cuDNN 可能未正确安装。
TensorFlow-gpu 没有运行,所以要么 要么tensorflow
有cuDNN
问题。任何想法都值得赞赏。