ubuntu matlab中安装MatConvNet工具箱(支持GPU,mexcuda报错问题)

2022-05-07 15:34:13 浏览数 (1)

很早以前就装过MatConvNet,但是最近机器重启时发现要用到这个工具箱的时候却发现找不到了,只能重新安装了。

MatConvNet  is a MATLAB toolbox implementing  Convolutional Neural Networks  (CNNs) for computer vision applications. It is simple, efficient, and can run and learn state-of-the-art CNNs. Many pre-trained CNNs for image classification, segmentation, face recognition, and text detection are available.

从上面介绍可以看出,MatConvNet是一个CNN的工具箱,提供了丰富的图像操作。

安装过程在官网上也有介绍,也比较简单,就三个步骤:

(1)下载软件包,并将软件工具箱的路径设置为matlab的当前工作路径。

(2)进行编译工具箱:

vl_compilenn

由于正常情况下,需要使用GPU来训练,因此需要在编译的过程中设置支持GPU。稍微复杂一些,设置编译过程如下:

vl_compilenn('enableGPU',true, 'cudaMethod', 'nvcc', 'cudaRoot','/usr/local/cuda')

否则在不支持GPU的情况下,对一些模型的训练加载容易导致mexcuda无法工作的问题。

(3)安装工具箱

vl_setupnn

根据以上步骤,即可使用matconvnet工具箱。

0 人点赞