详细流程安装深度学习框架

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  

猜你喜欢

转载自blog.csdn.net/u014515463/article/details/81484315
今日推荐