My CUDA is version 5.5 and ubuntu version 14.04, and I have all the prerequisites I need, I have also install g++ 4.6. I have git clone the lastest caffe code.
Here is my Makefile.config
USE_CUDNN := 1
CUSTOM_CXX := g++
CUDA_DIR := /usr
CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
-gencode arch=compute_20,code=sm_21 \
-gencode arch=compute_30,code=sm_30 \
-gencode arch=compute_35,code=sm_35 \
BLAS := atlas
PYTHON_LIB := /usr/lib
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
DISTRIBUTE_DIR := distribute
DEBUG := 1
TEST_GPUID := 0
In my Makefile.
I have updated
ifeq ($(LINUX), 1)
CXX ?= /usr/bin/g++
to
ifeq ($(LINUX), 1)
CXX ?= /usr/bin/g++-4.6
But when I compile with make
I got a error,
./include/caffe/util/cudnn.hpp:5:19: fatal error: cudnn.h: No such file or directory
#include <cudnn.h>
and when I compile with cmake
I got a error
nvcc fatal : Unsupported gpu architecture 'compute_50'
CMake Error at caffe_cu_generated_im2col.cu.o.cmake:206 (message):
Error generating
/home/sizhexi/caffe/build/src/caffe/CMakeFiles/caffe_cu.dir/util/./caffe_cu_generated_im2col.cu.o
I am confused that in the Makefile.config, I have already removed the lines below
-gencode arch=compute_50,code=sm_50 \ -gencode arch=compute_50,code=compute_50 |
|
|