【记录】 编译matconvnet on ubuntu16.04 with cuda 9.0
最近需要用到matconvnet 在Ubuntu16.04下。 因为TensorFlow 1.6 支持了CUDA 9.0 所以新机器就直接装了9.0 但是编译matconvnet 时遇到了一些问题 特此记录一下。 1. Error using mex nvcc fatal : Unsupported gpu architecture 'compute_20' Solution: 这个是因为cuda 8 之后不支持compute_20 了,最低也是compute_30了。 所以需要将vl_compilenn.m中的以下代码进行修改 opts.defCudaArch = [... '-gencode=arch=compute_20,code=\"sm_20,compute_20\" '... '-gencode=arch=compute_30,code=\"sm_30,compute_30\"']; 我用的是GTX1080TI, 此处我修改成 opts.defCudaArch = [... '-gencode=arch=compute_30,code=\"sm_30,compute_30\" '... '-gencode=arch=compute_50,code=\"sm_50,compute_50\"']; 同时还需要将 matconvnet/matlab/src/config