CUDA和CUDNN的坑

CUDA和CUDNN的坑一直报这个错,网上的博客答案也各种各样。但最简单的方法

是在caffe的训练文件中的每个卷积层中加enjine:CAFFE;即不适用cudn的计算引擎,使用caffe自带的,但是对于我非常深的网络

一个个手写添加简直崩溃,想想还是硬着头皮重新配环境吧,人生的坑得自己走一边啊....

https://blog.csdn.net/pkuhyz/article/details/82427384

首先显卡的型号,驱动,CUDA ,CUDNN各种版本匹配,所以真的很痛苦。

首先cmd行下查看默认的版本,尤其的电脑上装了四个版本的CUDA,

CUDA和CUDNN的坑

发现我的默认的版本是9.0,而我的caffe的配置文件时8.0

CUDA和CUDNN的坑CUDA和CUDNN的坑

但是我的电脑的path下8.0也有啊,只不过电脑的默认版本是9.0,难道必须与电脑默认的版本非得一直???

CUDA和CUDNN的坑

cudnn:将其解压,将对应的文件夹放入到CUDA下,CUDA安装路径是默认的

CUDA和CUDNN的坑

注:关于这些包的下载 ,我就不细说明了。

现在就是开始重新编译caffe,先编译libcaffe,这一个项目,我的vs13必须将其并行build的文件个数设置为1,才不会编译出错

编译完了libcaffe,接下来就是整个caffe的工程编译,就结束啦,收工,接下来就是胆战心惊的测试,看看到底解决了问题吗??

 

先用脚本看看:https://www.jianshu.com/p/9bdeb033e765

CUDA和CUDNN的坑

CUDA和CUDNN的坑

CUDA和CUDNN的坑

 

Ok了,在训练自己的网络看看,还有没有之前的错误呢,嗯嗯,真的没有问题了,强迫症的我终于又踩完一个坑 ,撒花。。。