Skip to content

Commit 6328b53

Browse files
committed
Makefile, Cmake and Travis script update.
1 parent 513fade commit 6328b53

File tree

4 files changed

+17
-9
lines changed

4 files changed

+17
-9
lines changed

CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ include(cmake/Summary.cmake)
1515
include(cmake/ConfigGen.cmake)
1616

1717
# ---[ Options
18-
caffe_option(CPU_ONLY "Build Caffe wihtout CUDA and OpenCL support" OFF) # TODO: rename to USE_CUDA
18+
caffe_option(CPU_ONLY "Build Caffe wihtout CUDA and OpenCL support" OFF)
1919
caffe_option(USE_CUDA "Build Caffe with CUDA support" ON)
2020
caffe_option(USE_GREENTEA "Build Caffe with OpenCL support" ON)
2121
caffe_option(USE_CUDNN "Build Caffe with cuDNN libary support" ON IF USE_CUDA AND NOT CPU_ONLY)

Makefile.config.example

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
USE_CUDA := 1
88

99
# Enable the OpenCL/Greentea backend
10-
USE_GREENTEA := 1
10+
USE_GREENTEA := 0
1111

1212
# Folder of the ViennaCL header-only library
1313
VIENNACL_DIR = ../ViennaCL
@@ -17,7 +17,7 @@ VIENNACL_DIR = ../ViennaCL
1717
# When both are disabled, GPUs won't work. CPUs always use CBLAS (Atlas, MKL or OpenBLAS).
1818
# The chosen BLAS library needs to be compiled and installed from source.
1919
# CLBLAS should be faster, especially on AMD cards.
20-
USE_CLBLAS := 1
20+
USE_CLBLAS := 0
2121
USE_VIENNACLBLAS := 0
2222

2323
# Enable or disable double precision support for OpenCL/Greentea

scripts/travis/travis_install.sh

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,17 +9,25 @@ MAKE="make --jobs=$NUM_THREADS"
99

1010
# This ppa is for gflags and glog
1111
add-apt-repository -y ppa:tuleu/precise-backports
12+
# This ppa is for boost 1.54
13+
add-apt-repository -y ppa:boost-latest/ppa
14+
# This ppa is for g++ 4.8
15+
add-apt repository -y ppa:ubuntu-toolchain-r/test
16+
1217
apt-get -y update
1318
apt-get install \
14-
wget git curl \
19+
g++-4.8 wget git curl \
1520
python-dev python-numpy \
1621
libleveldb-dev libsnappy-dev libopencv-dev \
17-
libboost-dev libboost-system-dev libboost-python-dev libboost-thread-dev \
22+
libboost-dev libboost-system-dev libboost-python-dev libboost-thread-dev libboost-python1.54-dev \
1823
libprotobuf-dev protobuf-compiler \
1924
libatlas-dev libatlas-base-dev \
2025
libhdf5-serial-dev libgflags-dev libgoogle-glog-dev \
2126
bc
2227

28+
update-alternatives
29+
--install /usr/bin/g++ g++ /usr/bin/g++-4.8 90
30+
2331
# Add a special apt-repository to install CMake 2.8.9 for CMake Caffe build,
2432
# if needed. By default, Aptitude in Ubuntu 12.04 installs CMake 2.8.7, but
2533
# Caffe requires a minimum CMake version of 2.8.8.
@@ -31,19 +39,19 @@ fi
3139

3240
# Install CUDA, if needed
3341
if $WITH_CUDA; then
34-
CUDA_URL=http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1204/x86_64/cuda-repo-ubuntu1204_6.5-14_amd64.deb
42+
CUDA_URL=http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1204/x86_64/cuda-repo-ubuntu1204_7.0-28_amd64.deb
3543
CUDA_FILE=/tmp/cuda_install.deb
3644
curl $CUDA_URL -o $CUDA_FILE
3745
dpkg -i $CUDA_FILE
3846
rm -f $CUDA_FILE
3947
apt-get -y update
4048
# Install the minimal CUDA subpackages required to test Caffe build.
4149
# For a full CUDA installation, add 'cuda' to the list of packages.
42-
apt-get -y install cuda-core-6-5 cuda-cublas-6-5 cuda-cublas-dev-6-5 cuda-cudart-6-5 cuda-cudart-dev-6-5 cuda-curand-6-5 cuda-curand-dev-6-5
50+
apt-get -y install cuda-core-7-0 cuda-cublas-7-0 cuda-cublas-dev-7-0 cuda-cudart-7-0 cuda-cudart-dev-7-0 cuda-curand-7-0 cuda-curand-dev-7-0
4351
# Create CUDA symlink at /usr/local/cuda
4452
# (This would normally be created by the CUDA installer, but we create it
4553
# manually since we did a partial installation.)
46-
ln -s /usr/local/cuda-6.5 /usr/local/cuda
54+
ln -s /usr/local/cuda-7.0 /usr/local/cuda
4755
fi
4856

4957
# Install LMDB

src/caffe/layers/cudnn_conv_layer.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ void CuDNNConvolutionLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top,
150150
// Synchronize the work across groups, each of which went into its own
151151
// stream, by launching an empty kernel into the default (null) stream.
152152
// NOLINT_NEXT_LINE(whitespace/operators)
153-
sync_conv_groupsCUDA_KERNEL(1, 1)();
153+
sync_conv_groups CUDA_KERNEL(1, 1)();
154154
}
155155
}
156156

0 commit comments

Comments
 (0)