Ubuntu 安装 tensorflow

Ubuntu 安装 tensorflow

1、安装 NVIDIA 显卡驱动

1.1、检测你的NVIDIA显卡型号和推荐的驱动安装型号

ubuntu-drivers devices

安装上面推荐的驱动程序：

sudo apt-get install nvidia-driver-455

1.2、安装完成后重启

sudo reboot

1.3、检查是否安装成功

nvidia-smi

---

2、安装 anaconda

2.1、安装

sudo bash Anaconda3-2020.11-Linux-x86_64.sh

按照提示安装即可，默认将安装在 /home/用户名 目录下，如需更改，可以自行指定。最后允许将 Anaconda 路径添加到 bashrc 中。

2.2、验证

li@li-911M:~/下载$ /home/li/anaconda3/bin/python
Python 3.8.5 (default, Sep  4 2020, 07:30:14) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> 

---

3、安装 pycharm

3.1、解压缩并找到 \bin 目录

sh ./pycharm.sh

就可以开始正常安装了！

3.2、创建快捷方式

1. 在 /usr/share/applications 创建一个文件：pycharm.desktop（touch，gedit，nano指令等）

   cd /usr/share/applications
   sudo gedit pycharm.desktop
   

2. 编辑这个文件，添加以下内容

   [Desktop Entry]
   Version=1.0
   Type=Application
   Name=Pycharm
   Icon=/home/mengdan/Downloads/pycharm-2018.2.1/bin/pycharm.png
   Exec=sh /home/mengdan/Downloads/pycharm-2018.2.1/bin/pycharm.sh
   MimeType=application/x-py;
   Name[en_US]=pycharm
   

   注：这里 Icon 和 Exec 的绝对路径，从该文件的属性里查看。

---

4、安装 cudn

下载地址：https://developer.nvidia.com/cuda-toolkit-archive

4.1、准备工作

准备工作：这里需要先卸载之前安装的显卡驱动和禁用 nouveau 驱动

卸载之前安装的显卡驱动：

sudo apt-get remove --purge nvidia*

禁用 nouveau 驱动：

#步骤一：
sudo vim /etc/modprobe.d/blacklist.conf

'''
在文本最后添加：
blacklist nouveau
options nouveau modeset=0
'''

# 步骤二：
# 更新系统
sudo update-initramfs -u

# 步骤三：验证nouveau是否已禁用
# 重启后，执行以下命令，如果没有屏幕输出，说明禁用nouveau成功：
'一定要重启'

lsmod | grep nouveau

4.2、安装 cudn

安装 cudn（这里会自动安装与之匹配的显卡驱动）

sudo sh cuda_11.0.2_450.51.05_linux.run

4.3、配置环境变量

在 ~/.bashrc 的最后加上以下信息：

sudo gedit ~/.bashrc

export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}

source ~/.bashrc

4.4、检查是否安装成功

使用 nvcc -V 来检查是否安装成功：

(base) li@li-911M:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Thu_Jun_11_22:26:38_PDT_2020
Cuda compilation tools, release 11.0, V11.0.194
Build cuda_11.0_bu.TC445_37.28540450_0

---

5、安装 cudnn

下载地址：https://developer.nvidia.com/rdp/cudnn-archive

5.1、解压缩

tar -zxvf cudnn-11.0-linux-x64-v8.0.4.30.tgz

5.2、拷贝文件

使用以下两条命令复制这些文件到 CUDA 目录下：

sudo cp cuda/lib64/* /usr/local/cuda/lib64/
sudo cp cuda/include/* /usr/local/cuda/include/

---

6、安装 tensorflow

在 anaconda 的解释器中安装 tensorflow：

/home/username/anaconda/bin/pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple

---

7、安装 tensorflow-gpu

/home/username/anaconda/bin/pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple