Removed NCCL support (it can be enabled by uncommenting the proper lines)

The newly released NCCL version 2 requires registration at the
NVIDIA developer webiste and a manual download. As this breaks
automated builds by AUR helpers, NCCL was disabled by default in
this package. It can still be enabled by uncommenting the proper
lines in the PKGBUILD file (instructions are at the top of the
PKGBUILD file).
master
Daniel Bermond 7 years ago committed by Peter Babič
parent 006d0d64c6
commit 2662fca159
  1. 5
      .SRCINFO
  2. 12
      PKGBUILD

@ -1,9 +1,9 @@
# Generated by mksrcinfo v8 # Generated by mksrcinfo v8
# Sun Aug 6 13:20:29 UTC 2017 # Sun Sep 3 15:40:18 UTC 2017
pkgbase = caffe pkgbase = caffe
pkgdesc = A deep learning framework made with expression, speed, and modularity in mind (gpu enabled) pkgdesc = A deep learning framework made with expression, speed, and modularity in mind (gpu enabled)
pkgver = 1.0 pkgver = 1.0
pkgrel = 5 pkgrel = 6
url = http://caffe.berkeleyvision.org/ url = http://caffe.berkeleyvision.org/
arch = x86_64 arch = x86_64
license = BSD license = BSD
@ -38,7 +38,6 @@ pkgbase = caffe
depends = python-pillow depends = python-pillow
depends = python-six depends = python-six
depends = openblas-lapack depends = openblas-lapack
depends = nccl
depends = python-leveldb depends = python-leveldb
depends = python-scikit-image depends = python-scikit-image
depends = python-pydotplus depends = python-pydotplus

@ -1,9 +1,13 @@
# Maintainer : Daniel Bermond < yahoo-com: danielbermond > # Maintainer : Daniel Bermond < yahoo-com: danielbermond >
# Contributor: Micah Chambers <micahc.vt@gmail.com> # Contributor: Micah Chambers <micahc.vt@gmail.com>
# NOTE:
# In order to build with NCCL support, uncomment the NCCL
# lines in 'depends' and 'preprare()'.
pkgname=caffe pkgname=caffe
pkgver=1.0 pkgver=1.0
pkgrel=5 pkgrel=6
pkgdesc="A deep learning framework made with expression, speed, and modularity in mind (gpu enabled)" pkgdesc="A deep learning framework made with expression, speed, and modularity in mind (gpu enabled)"
arch=('x86_64') arch=('x86_64')
url="http://caffe.berkeleyvision.org/" url="http://caffe.berkeleyvision.org/"
@ -18,8 +22,8 @@ depends=(
# AUR: # AUR:
# required: # required:
'openblas-lapack' 'openblas-lapack'
# not required but enabled in build: # not required:
'nccl' # 'nccl'
# python: # python:
'python-leveldb' 'python-scikit-image' 'python-pydotplus' 'python-leveldb' 'python-scikit-image' 'python-pydotplus'
# NOTE: # NOTE:
@ -42,7 +46,7 @@ prepare() {
sed -i '/USE_CUDNN/s/^#[[:space:]]//g' Makefile.config sed -i '/USE_CUDNN/s/^#[[:space:]]//g' Makefile.config
# enable NCCL acceleration switch # enable NCCL acceleration switch
sed -i '/USE_NCCL/s/^#[[:space:]]//g' Makefile.config # sed -i '/USE_NCCL/s/^#[[:space:]]//g' Makefile.config
# strictly enable I/O dependencies # strictly enable I/O dependencies
sed -i '/USE_OPENCV/s/^#[[:space:]]//;/USE_OPENCV/s/0/1/' Makefile.config sed -i '/USE_OPENCV/s/^#[[:space:]]//;/USE_OPENCV/s/0/1/' Makefile.config

Loading…
Cancel
Save