caffe 环境安装问题 找不到cudart64_90.dll

安装的是github上happynear的caffe版本,主要是方便,地址是https://github.com/happynear/caffe-windows;

按照作者的文档修改工程属性,下载好第三方依赖库即可。

编译没什么问题,就是在运行caffe的程序时,提示缺少cudart64_90.dll;

我的环境是win10, cuda9.2+cudnn7.1.

happynear大神的回复是  caffe所需的nccl库是作者自己编译的,依赖cuda9.0;所以在我的环境上会提示缺少cudart64_90.dll

只需要下载nccl工程自己编译下即可。地址https://github.com/happynear/nccl

打开里面的windows文件夹,修改工程属性文件nccl.vcxproj,找到里面两个cuda版本号设置的地方,改为9.2(我自己的环境)

重新编译生成nccl.dll和nccl.lib ,替换caffe文件夹中thirdparty里的nccl.dll和lib。注意替换全,bins文件夹和nccl文件夹都替换.

你可能感兴趣的:(人工智能)