ubuntu配置深度学习环境

image

目前最新的 Ubuntu Server 16.04.3 LTS 系统为例，介绍包括 CUDA、CudNN 软件以及 TensorFlow、Keras 等 Python 库的安装方法。

下面是需要安装的软件列表。

• Cuda
• CudNN
• TensorFlow, Pytorch
• OpenCV,Caffe
• Matlab

前期准备： sudo apt-get update， 找到清华大学镜像站，下好anaconda并安装，然后将anaconda的python添加到环境变量

然后下载到NVIDIA下载cuda

image

一个坑： 必须关闭图形界面，不然会报错

sudo service lightdm stop

image

cuda安装好

image

下面安装cuDNN， 用来加速深度学习框架

到英伟达官网下载好压缩包，然后解压缩。将cuda/lib64的文件复制到 /usr/local/cuda/lib64中，cuda/include/cudnn.h 复制到 /usr/local/cuda/include

然后更新libcudnn.so.7 连接， 然后ldconfig --> 搜索出可共享的动态链接库（格式如lib.so）,进而创建出动态装入程序(ld.so)所需的连接和缓存文件。

sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig -v

到NVIDIA给的sample 里面测试一下

cd samples && make

等待一会之后，编译完成。

./bandwidthtest

image