最近学长布置的任务是对于不同QP下的半像素插值进行分别训练,并且与单独训练一个网络用于所有QP的效果进行比较。但是实验室最近资源也有点紧,所以想利用自己的电脑也配一个GPU版caffe,就可以在寝室干活了。但是在根据其他博客上面的指引,出现了很多没见过的问题,那就在这里记下自己的过程以及遇到的一些问题吧
一、环境
win10
vs2013 community(ultimate应该也是一样的)
i7-6700HQ+GTX965M
二、CUDA以及cudnn
其实今天被这里坑的最惨,最终发现失败的原因就在于cudnn
CUDA
首先是安装CUDA,网站上可以直接下载,我用的是学长直接提供的cuda_8.0.61版本,目前官网默认下载的版本已经到了9.2,但是今天的经历告诉我,最新的很可能不是最好的,甚至可能是不能work的!
链接:https://pan.baidu.com/s/1_7bGATxfZiOPEOfKp183Gw 密码:ca9m
具体步骤:我们手头能够拿到的是.exe文件,直接不停的下一步就行了,然后安装目录尽量不要改吧,就用默认的就行。然后路径最好是英文的,中文路径以后可能会出问题。(这一步不会有什么问题,就不放图了)
cudnn
接着是安装cudnn,这个从官网下载的话得到的是一个压缩文件,但是需要注册一个nvidia账号,所以下面也提供了百度云。如果要网上自行下载的话一定要选择适配自己CUDA版本的cudnn。另外一个事情,也是今天一直失败的原因,就是忽视了cudnn的版本。后来从BLVC的github上面看到cudnn应该使用v5版本!我之前用了v6,所以一直出错,改成正确版本之后果然就成功了~
链接:https://pan.baidu.com/s/1LtYW8k9RRUNJDySlaN4N9w 密码:p0vd
具体步骤:下载之后得到的是一个压缩包,有一个cuda文件夹,里面有bin,include,lib三个文件夹。
首先解压,假设我解压到的位置是C:\lab\cudnn-8.0-windows10-x64-v5.0-ga,那么这个位置下面就会有一个cuda文件夹,里面有这三个文件夹。
然后打开本机的环境变量,找到系统变量中的CUDA_PATH
打开后面的那一个位置
可以看到,这里也有lib,bin,include三个文件夹,于是我们需要将之前解压得到的cuda文件夹里面的三个文件夹中的内容拷贝到这里的对应的文件夹中(例如:cuda/bin/cudnn64_5.dll拷贝到上图中的bin里面),总共有三个文件需要拷贝,其中第三个在cuda/x64中。
以上,我们的CUDA和cudnn就装好了~
三、caffe源代码下载+编译
源码位置:https://github.com/microsoft/caffe
熟悉git的话直接git clone就行了,不会的话也没事,点击绿色按钮之后download zip就行了,得到一