ubuntu16.04安装深度学习框架
1. 安装NVIDIA GPU 驱动(ubuntu16.4)
依次选择system settings -> Software & Updates->Additional Drivers
NVIDIA Corporation: Unknown
using NVIDIA binary driver -version 384.130 from nvidia-384
点击 Apply
2. 更新系统源
依次选择system settings -> Software & Updates-> ubuntu software
Download from http://mirrors.aliyun.com/ubuntu
3.下载 cuda9.0&cudnn&Anaconda&pycharm:
4.安装 cuda9.0
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
一直输入 y
5. 设置CUDA环境变量
安装vim
sudo apt-get install vim
vim ~/.bashrc
输入Go 切换到编辑模式 (将下面文字复制上去:)
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
输入 ESC 切换模式(ESC为键盘左上角按键)
输入 :wq 回车退出Vim
source ~/.bashrc
6. 安装 cuDNN7.0
解压下载的cudnn文件
tar -zxvf cudnn-9.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
7. 更新 pip source
mkdir ~/.config/pip/
vim ~/.config/pip/pip.conf
输入 a 切换编辑模型(将下面文字复制上去:)
[global]
trusted-host = mirrors.aliyun.com
index-url = http://mirrors.aliyun.com/pypi/simple
输入 ESC :wq 回车保存退出(ESC为键盘左上角按键)
8. 安装 anaconda3
sh Anaconda3-5.1.0-Linux-x86_64.sh
输入 yes 同意协议及设置环境变量等
9. 安装 tensorflow-gpu
pip install tensorflow-gpu==1.5.1
若无GPU则
pip install tensorflow==1.5.1
10. 安装 keras
pip install keras
11. 安装 opencv
conda install opencv
pip install opencv-python
12. 安装 pytorch
pip install torch
pip install torchvision