Ubuntu 安装 tensorflow

Ubuntu 安装 tensorflow

1、安装 NVIDIA 显卡驱动

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

ubuntu-drivers devices

Ubuntu 安装 tensorflow_第1张图片

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

sudo apt-get install nvidia-driver-455

1.2、安装完成后重启

sudo reboot

1.3、检查是否安装成功

nvidia-smi

Ubuntu 安装 tensorflow_第2张图片


2、安装 anaconda

2.1、安装

sudo bash Anaconda3-2020.11-Linux-x86_64.sh

Ubuntu 安装 tensorflow_第3张图片

按照提示安装即可,默认将安装在 /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 目录

Ubuntu 安装 tensorflow_第4张图片

Ubuntu 安装 tensorflow_第5张图片

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

你可能感兴趣的:(tensorflow深度学习,深度学习)