Ubuntu安装anaconda,tensorflow,keras,pytorch

仰望大牛的小清新:http://www.cnblogs.com/luruiyuan/

本文原网址:http://www.cnblogs.com/luruiyuan/p/6660142.html

经验证:cuDNN v6 + tensorflow 1.3.0 + keras 2.0.5 不会死核心

tensorflow 和keras 请用pip安装 ,conda安装的话会安装一大堆


一,安装ubuntu16.04

单击“创建新分区表",点击的“+”创建4个主要的基础分区(这里之前未分配的100G就是给ubuntu系统的100G),按以下参数设置4个主要的基础分区:

40G     主分区     空间起始位置     Ext4日志文件系统           /    

10G    逻辑分区         空间起始位置     交换空间      /swap

500MB     逻辑分区    空间起始位置     Ext4日志文件系统    /boot

15G          逻辑分区     空间起始位置    Ext4日志文件系统     /temp

剩余的空间     逻辑分区    空间起始位置     Ext4日志文件系统    /home

(当然这里还可创建自己的其他分区如/123分区,只要设置为逻辑分区、空间起始位置、Ext4日志文件系统即可,新手强烈建议忽略此步,按上面设置即可)

安装启动引导设备的参数选择:与/boot所在的编号一致。设置好后安装即可,之后的选项(所在地点、键盘布局、电脑名称、开机密码等)正常进行直到安装完毕,大功告成。

打开windows盘

sudo ntfsfix /dev/sda2


1.下载 CUDA 8.0

地址:https://developer.nvidia.com/cuda-downloads

安装如下版本:

Ubuntu安装anaconda,tensorflow,keras,pytorch_第1张图片

2.下载 cuDNN v6 (原教程里是V5,但发现容易核心死掉)

经验证:cuDNN v6 + tensorflow 1.3.0 + keras 2.0.5 不会死核心

tensorflow 和keras 请用pip安装 ,conda安装的话会安装一大堆

地址: https://developer.nvidia.com/cudnn

这里需要先 登录/注册 后才能下载

Ubuntu安装anaconda,tensorflow,keras,pytorch_第2张图片

3.安装 NVIDIA 驱动:

首先打开 terminal,输入一下指令,更新应用列表

sudo apt-get update

多亏了 Linux 社区的强大支持,我们可以用很简洁的方式安装驱动:

按下 win 按键,打开菜单,如图:

然后在上方的 type to search 中输入: additional drivers 打开 "additional drivers —— 附加驱动",然后选择 与自己显卡匹配的 NVIDIA 驱动,我这里是 

using nvidia binary drivers (375)

然后选择 应用更改,这里安装可能会失败,此时你可以继续选择应用更改,多安装几次即可

安装完成后,选择 restart,一定要重启

4. gcc 降版本

CUDA8.0 不支持 gcc 5.0 及以上的编译器,而系统自带的是5.4及以上版本,因此我们需要降版本,否则会在后面报错

在terminal中输入如下指令,将 gcc 版本降到 4.9

sudo apt-get install g++-4.9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
sudo update-alternatives --set cc /usr/bin/gcc
sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo update-alternatives --set c++ /usr/bin/g++ 

5.安装 CUDA 8.0

cd /media/你的用户名/TOSHIBA\ EXT/alu/CUDA/ # 进入cuda 所在文件夹
# 根据官网上的提示安装 cuda 8.0
sudo dpkg -i cuda-repo-ubuntu1604-8-0-rc_8.0.27-1_amd64​.deb
sudo apt-get update
sudo apt-get install cuda​

6. 安装 cuDNN

cd  cd /media/你的用户名/TOSHIBA\ EXT/alu/CUDA/ # 进入 cuDNN 安装文件的所在路径
tar xvzf cudnn-8.0-linux-x64-v6.0.tgz # 解压
sudo cp cuda/include/cudnn.h /usr/local/cuda/include # 复制到 include 中
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 # 复制到 lib64 中
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* # 讲头文件复制进去

7. 配置CUDA环境变量

开启 GPU 支持:

根据官网教程

Ubuntu安装anaconda,tensorflow,keras,pytorch_第3张图片

我们在terminal中键入下列命令:

sudo gedit ~/.bash_profile # 打开.bash_profile 这是用户的环境变量,不是全局的

然后在打开的文本末尾加入:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

保存并关闭后,输入下列命令使环境变量生效:

source ~/.bash_profile # 使被更改的环境变量生效

安装完成后,一定要通过如下两条命令对驱动进行检验:

1. NVIDIA的设置界面

nvidia-settings # 打开 NVIDIA 设置界面

这条指令打开的界面如下:

Ubuntu安装anaconda,tensorflow,keras,pytorch_第4张图片

2. NVIDIA GPU 列表

nvidia-smi

这条指令会在 terminal 中产生GPU列表,如,我这里只有一个GPU

网上有些人抄别人的博客,说 nvcc -V 就可以验证,经过我实测,存在 nvcc -V 正常输出但是驱动仍安装失败的现象,因此,上述验证方法是不可信的

8.安装 python3.5.2

由于tensorflow1.0 对于 python3 支持更好,并且目前只支持python3.5.2,因此我们选择 python 3.5.2。

使用 Anaconda3-4.2.0-Linux-x86_64.sh 进行安装,地址如下:

https://repo.continuum.io/archive/

命令:bash 文件名

安装完成后,添加环境变量,将其设置为默认的 python 解释器

首先打开环境变量的文件

gedit ~/.bashrc

然后在文件末尾加入 anaconda3 的路径

export PATH=/home/你的路径/anaconda3/bin:$PATH

如果有了,就不要加了

最后使我们的改动生效

source ~/.bashrc

这样,我们在terminal中输入 python 就会默认打开 anaconda3

这样我们就可以安心使用python3了。

9.安装 keras 和 tensorflow

有了上述安装过程,我们系统中默认的pip将会是anaconda3中自带的pip,这样我们只需要使用pip即可安装 keras 和 tensorflow 到 anaconda 中。

执行如下命令:

pip install tensorflow-gpu keras # 安装 gpu 版本的 tensorflow 和 keras

安装完成后,我们使用如下命令,即可检验是否成功:

python -c "import keras"

如果看到如下输出,就说明安装成功

当然了,我这里安装的 cuDNN 由于版本过高,暂时不能被pip安装的tensorflow所支持,如果改为 cuDNN v5 就能够正常支持了。

希望自己的这篇文章能对像我一样的新手有所帮助。

9,为了使tensorflow运行还需要安装libcupti-dev库

libcupti-dev库,即NVIDIA CUDA配置文件工具界面。此库提供高级分析支持。要安装此库,请发出以下命令:

 
    
$ sudo apt-get install libcupti-dev 10,安装pytorch

http://pytorch.org/
conda install pytorch torchvision cuda80 -c soumith

安装指定版本

https://ptorch.com/news/37.html

conda install pytorch=0.1.12 -c soumith

11,重复安装cuDNN的解决办法


         1.根据提示,应该时Cudnn版本的问题,本身装的时CUDA7.5,如果原先有安装cudnn,需要先把拷贝的几个文件删除:
        cd  /usr/local/cuda/
        cd  include
        sudo rm cudnn.h
        cd lib64 
        sudo  rm libcudnn* 
如果没有装过cudnn,则不需要执行此步骤。
      2. 然后下载最新版本的Cudnn,注意一定要最新版本,我用的cuda7.5,先是装了cudnn v5.0,发现还是不行,后来装了v5.1就好了。
      登陆NVIDIA 网站注册后就能下载:https://developer.nvidia.com/rdp/cudnn-download
      最新版本是: cuDNN v5.1 Library for Linux

     3.下载后,使用以下命令解压并拷贝到cuda就ok了:
       tar -zxf cudnn-7.5-linux-x64-v5.1-prod.tgz
       cd cuda
       sudo cp lib64/* /usr/local/cuda/lib64/
       sudo cp include/cudnn.h /usr/local/cuda/include/

参考资料:

[1]: ubuntu16.04下安装TensorFlow(GPU加速)----详细图文教程 

[2]: Ubuntu16.04+cuda8.0+caffe安装教程 


你可能感兴趣的:(环境搭建)