原文标题:Setting up a Deep Learning Machine from Scratch (Software) 原文链接:https://github.com/saiprashanths/dl-setup 译者:刘翔宇 审校:赵屹华 责编:周建丁(zhoujd@csdn.net)
这是一篇为机器搭建深度学习研究环境的详细指南,包括驱动程序、工具和各种深度学习框架的安装指导。在64位Ubuntu 14.04的机器上使用Nvidia Titan X进行测试。
还有一些有类似目的的指南。一些内容有限,而另外一些则不是最新的。该指南基于(有些部分是复制来的):
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install build-essential sudo apt-get autoremove
安装git
sudo apt-get install git
lspci | grep -i nvidia
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-352
sudo shutdown -r now
cat /proc/driver/nvidia/version
sudo dpkg -i cuda-repo-ubuntu1404*amd64.deb
sudo apt-get update
sudo apt-get install cuda
echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrcecho 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrcsource ~/.bashrc
nvcc -V
sudo shutdown -r now
/usr/local/cuda/bin/cuda-install-samples-7.5.sh ~/cuda-samplescd ~/cuda-samples/NVIDIA*Samples
make -j $(($(nproc) + 1))
注意:(-j $(($(nproc) + 1)))命令使用你机器上的核心数并行执行,所以编译会更快
bin/x86_64/linux/release/deviceQuery
cd ~/Downloads/tar xvf cudnn*.tgz
cd cuda
sudo cp */*.h /usr/local/cuda/include/sudo cp */libcudnn* /usr/local/cuda/lib64/sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
sudo apt-get install python-pip python-dev sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.8.0-cp27-none-linux_x86_64.whl
python>>> import tensorflow as tf>>> exit()
mkdir ~/gitcd ~/git
git clone https://github.com/xianyi/OpenBLAS.gitcd OpenBLAS
make FC=gfortran -j $(($(nproc) + 1))sudo make PREFIX=/usr/local install
echo 'export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH' >> ~/.bashrc
sudo apt-get install -y libfreetype6-dev libpng12-devpip install -U matplotlib ipython[all] jupyter pandas scikit-image
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compilersudo apt-get install --no-install-recommends libboost-all-devsudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
cd ~/git
git clone https://github.com/BVLC/caffe.gitcd caffecp Makefile.config.example Makefile.config
sed -i 's/# USE_CUDNN := 1/USE_CUDNN := 1/' Makefile.config
sed -i 's/BLAS := atlas/BLAS := open/' Makefile.config
sudo pip install -r python/requirements.txt
make all -j $(($(nproc) + 1))
make test -j $(($(nproc) + 1))
make runtest -j $(($(nproc) + 1))
make pycaffe -j $(($(nproc) + 1))
echo 'export CAFFE_ROOT=$(pwd)' >> ~/.bashrcecho 'export PYTHONPATH=$CAFFE_ROOT/python:$PYTHONPATH' >> ~/.bashrcsource ~/.bashrc
ipython>>> import caffe>>> exit()
sudo apt-get install python-numpy python-scipy python-dev python-pip python-nose g++ python-pygments python-sphinx python-nosesudo pip install Theano
python>>> import theano>>> exit()
sudo pip install keras
git clone https://github.com/torch/distro.git ~/git/torch --recursive
cd torch; bash install-deps;./install.sh
sudo apt-get install software-properties-commonsudo add-apt-repository ppa:x2go/stable
sudo apt-get update
sudo apt-get install x2goserver x2goserver-xsession
sudo apt-get update
sudo apt-get install -y xfce4 xfce4-goodies xubuntu-desktop
hostname -I