MatConvNet+Matlab2017a+CUDA8.0安装

MatConvNet+Matlab2017a+CUDA8.0安装

折腾了一天半终于装好了win10下的TensorFlow-GPU版,在这里做个记录。

准备安装包:

visual studio 2015;

Anaconda3-4.2.0-Windows-x86_64

pycharm-community

CUDA:cuda_8.0.61_win10;下载时选择 exe(local)

CUDA补丁:cuda_8.0.61.2_windows

cuDNN:cudnn-8.0-windows10-x64-v6.0;如果你安装的TensorFlow版本和我一样1.3,请下载cuDNN v6.0 for CUDA 8.0 (不要问我为什么知道....)

开始:

1.安装visual studio2015 可以只安装 Visualc++部分

2.安装CUDA:

按提示安装,先安装cuda_8,再安装补丁;

装完后在cmd里查看版本号:nvcc -V 

 

3.安装cuDNN库:

把解压文件放置到CUDA的相关文件夹里:(懒得打字了)

==============》》  

 

安装过程参照MatConvNet官网给出的步骤:

http://www.vlfeat.org/matconvnet/install/

1.安装CUDA8.0+cudnn6.0 参见之前的博客

2.安装MatConvNet:

下载地址:matconvnet-1.0-beta25.tar.gz

安装步骤:

我先安装了CPU版,然后安装了GPU版:

CPU版:

1.解压matconvnet到自己的目录。

2.打开MatLab,把matlab添加到MATLAB的搜索路径中。

3.

1
2
> cd <MatConvNet>
> addpath matlab

4.

> vl_compilenn

如果一切顺利(忽略同名警告),编译完后就可以使用cpu版了。

5.测试:

> vl_testnn

测试结果:

 

GPU版:

前提要求:

安装步骤:

1. 如果你的电脑了里面只有一个CUDNN版本,并且与matlab版本相匹配,直接用如下代码进行编译:

vl_compilenn('enableGpu', true)

 

安装后的测试:

 

 测试结果:

posted @ 2018-03-22 16:23  菜鸡一枚  阅读(640)  评论(0)    收藏  举报