Ubuntu18.04安装配置Tensorflow-gpu

安装Anaconda3

  • 在清华镜像站下载sh文件,目前最新版本是Anaconda3-2019.07-Linux-x86_64.sh
    网址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
  • 安装:打开终端,cd到下载文件位置
bash Anaconda3-2019.07-Linux-x86_64.sh
上面为下载的文件名
  • 检查环境变量:安装完成后在终端输入
conda --version

若出现conda版本信息,则环境变量配置成功,提示没有指令的话,需要配置环境变量。

  • 配置环境变量:在终端输入
sudo gedit ~/.bashrc

在末尾输入
export PATH="/home/lone/anaconda3/bin:$PATH"
上面的/home/lone/anaconda3为安装anaconda3路径

source ~/.bashrc

配置结束之后重新检查一下环境变量是否成功配置
####添加conda虚拟环境

  • 创建虚拟环境
conda create --name yourname python=3.7
yourname为自定义环境名 
  • 激活环境
source activate yourname
yourname为环境名

激活后在终端左侧出现(yourname),即为激活成功

  • 在环境中安装包(需要激活环境)
两种方式
conda install package
pip install package
package为包名
  • conda添加国内镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

pip下载比较慢的话可以修改源地址,手动修改pip源方法

 pip install package -i https://pypi.doubanio.com/simple
 package为包名
  • 在pycharm配置conda-python
点击顺序:file -> Settings -> Project:project名 -> Project InterPreter -> 右边小齿轮 -> add -> Existing environment -> 选择路径:conda位置/envs/虚拟环境名/bin/python3.7 -> 勾选Make available to all project 

配置显卡驱动

  1. Ubuntu添加阿里云源(如果没有添加的话) 网址:https://www.cnblogs.com/nixingguo/p/9017805.html
  2. 打开"软件与更新" 如图配置

Ubuntu18.04安装配置Tensorflow-gpu_第1张图片

点击关闭,更新缓存(需要联网),之后再打开"软件与更新",点击附加驱动,选择NVIDIA驱动安装,中间可能需要设定boot密码,输入12345678即可。

  1. 禁用nouveau驱动:https://blog.csdn.net/qq805934132/article/details/82909759
  2. 重启电脑,进入BIOS关闭安全启动后再启动系统即可
  3. 重启后进入终端输入:nvidia-smi
    出现显卡信息即成功安装。

安装cudatoolkit&cudnn

conda install cudatoolkit=10.0
conda install cudnn=7.6.0

以上两个版本号和Tensorflow2.0版本兼容,修改可能会不匹配导致bug

安装Tensorflow2.0

pip install tensorflow-gpu==2.0.0

你可能感兴趣的:(Ubuntu18.04安装配置Tensorflow-gpu)