MatConvNet+Matlab2017a+CUDA8.0安装
折腾了一天半终于装好了win10下的TensorFlow-GPU版,在这里做个记录。
准备安装包:
visual studio 2015;
Anaconda3-4.2.0-Windows-x86_64;
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)
安装后的测试:
测试结果:
请发表评论