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. 打开"软件与更新" 如图配置

点击关闭,更新缓存(需要联网),之后再打开"软件与更新",点击附加驱动,选择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
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值