Caffe2
Price of Caffe2: free
Version: 0.8.1
CUDNN_URL="http://developer.download.nvidia®.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz"
wget ${CUDNN_URL}
sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local
rm cudnn-8.0-linux-x64-v5.1.tgz && sudo ldconfig
sudo apt-get update
sudo apt-get install -y --no-install-recommends \
build-essential \
cmake \
git \
libgoogle-glog-dev \
libgtest-dev \
libiomp-dev \
libleveldb-dev \
liblmdb-dev \
libopencv-dev \
libopenmpi-dev \
libsnappy-dev \
libprotobuf-dev \
openmpi-bin \
openmpi-doc \
protobuf-compiler \
python-dev \
python-pip
sudo pip install \
future \
numpy \
protobuf
# for Ubuntu 14.04
sudo apt-get install -y --no-install-recommends libgflags2
# for Ubuntu 16.04
sudo apt-get install -y --no-install-recommends libgflags-dev
git clone --recursive https://github.com/pytorch/pytorch.git && cd pytorch
git submodule update --init
mkdir build && cd build
5. Configure the assembly.
cmake ..
If you encounter errors during installation at the "Linking CXX executable ../bin/mpi_test" stage, then run the command:
cmake .. -DUSE_MPI=OFF
6. Compile and install Caffe2:
sudo make install
7. Verify Caffe2.
cd ~ && python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
If Caffe2 was installed successfully, then you should see the message "Success".
Category
Machine learning
Industry
Automotive, Aerospace, Electronic, Biotechnology, Medicine, Finance
License
Opensource/Free