系统环境ubuntu16.04带GPU显卡
1. 显卡驱动安装
第一步安装NVIDIA驱动
命令行终端敲入以下命令安装驱动
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt-get update
$ sudo apt-get install nvidia-384
$ sudo apt-get install mesa-common-dev
$ sudo apt-get install freeglut3-dev
$ sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev libglu1-mesa-dev
执行完上述后,重启(reboot)
然后输入以下命令进行测试
$ nvidia-smi
2.安装CUDA8.0包
到https://developer.nvidia.com/cuda-80-ga2-download-archive网上上下载
下载完cuda8.0包后,执行如下语句,先关闭图形界面,然后再运行runfile文件
$ sudo service lightdm stop
$ sudo sh cuda_8.0_36_linux.run
执行后会有一系列提示让你确认,但是注意,有个让你选择是否安装nvidia375驱动时,一定要选择否,因为前面我们已经安装了更加新的nvidia384,所以这里不要选择安装。其余的都直接默认或者选择是即可
安装成功后会出现如下界面:
然后进行环境变量配置
打开配置文件~/.bashrc
$ gedit ~/.bashrc
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
进行动态库配置
$ sudo gedit /etc/ld.so.conf.d/cuda.conf
在打开文件后加入
/usr/local/cuda-8.0/lib64
然后执行如下命令刷新系统动态库搜索缓存,使链接立即生效
$ sudo ldconfig
3.安装tensorflow
可以到tensorflow的官方安装网站参考https://www.tensorflow.org/install/install_linux
NVIDIA requirements to run TensorFlow with GPU support
Installing with native pip
$ sudo pip install numpy scipy
$ sudo apt-get install libcupti-dev
$ sudo apt-get install python-pip python-dev
$ sudo pip install tensorflow-gpu
测试tensorflow的程序
$ ipython
通过深度卷积网络,把移动设备上的照片提高到单反级别画质
开源工程https://github.com/aiff22/DPED
把该工程git clone 下来
$ git clone https://github.com/aiff22/DPED.git
进入到DPED工程目录下,测试
$ cd DPED
$ python test_model.py model=iphone_orig test_subset=full resolution=orig use_gpu=true
具体训练和测试该工程的github上有具体的例子