1. 程式人生 > >ubuntu16.04 caffe 安裝

ubuntu16.04 caffe 安裝

我的配置:ubuntu 16.04 + cuda9.0  + cudnn7.1  + opencv3.3.0

在自己實際安裝過程中,遇到了些問題,在此記錄下:

1、根據自己cuda的安裝版本,將Makefile.config檔案做相應的更改,即按照如下要求將不需要的部分註釋掉。

# CUDA architecture setting: going with all of them. # For CUDA < 6.0, comment the *_50 through *_61 lines for compatibility. # For CUDA < 8.0, comment the *_60 and *_61 lines for compatibility. # For CUDA >= 9.0, comment the *_20 and *_21 lines for compatibility.

2、Check failed: status == CURAND_STATUS_SUCCESS (201 vs. 0) CURAND_STATUS_LAUNCH_FAILURE