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、创建快捷方式
-
在 /usr/share/applications 创建一个文件：pycharm.desktop（touch，gedit，nano指令等）
cd /usr/share/applications sudo gedit pycharm.desktop -
编辑这个文件，添加以下内容
[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