在ubuntu14.04下配置caffe主要有以下步骤:
NVIDIA驱动以及cuda和cudnn的安装配置 Intel MKL的安装配置MATLAB的安装配置OpenCV的安装配置Anaconda以及pycuda的安装配置caffe的安装编译
本文介绍anaconda以及pycuda的安装配置
0.确保cuda已正确安装并且环境变量设置正确
注意:需要设置$CUDA_ROOT为你的cuda安装路径,添加$CUDA_ROOT/bin到$PATH中。
参考:在~/.bashrc中添加:
export CUDA_ROOT=/usr/local/cuda:$CUDA_ROOT
export PATH=/usr/local/cuda/bin:$PATH
1.安装anaconda
下载anaconda,选择linux Python2.7版本:https://www.continuum.io/downloads
执行:
bash Anaconda2-4.2.0-Linux-x86_64.sh
按照默认选项一步步进行安装即可。
anaconda可提供绝大多数python依赖包,包括numpy等。
2.安装boost C++
sudo apt-get install libboost-all-dev
3.安装pycuda
安装必要的库:
sudo apt-get install build-essential python-dev python-setuptools libboost-python-dev libboost-thread-dev -y
下载pycuda:https://pypi.python.org/pypi/pycuda
解压:
tar xzvf pycuda-VERSION.tar.gz
make及安装:
cd pycuda-VERSION
./configure.py --cuda-root=/usr/local/cuda --cudadrv-lib-dir=/usr/lib/x86_64-linux-gnu --boost-inc-dir=/usr/include --boost-lib-dir=/usr/lib --boost-python-libname=boost_python --boost-thread-libname=boost_thread --no-use-shipped-boost
make -j 4
sudo python setup.py install
sudo pip install .
注意:上面的命令是针对ubuntu14.04 64位python2.x版本的,若是其他版本,请参阅官方wiki。
转载请注明原文地址: https://ju.6miu.com/read-676353.html