ubuntu系统使用Anaconda安装tensorflow-gpu环境

ubuntu系统使用Anaconda安装tensorflow-gpu环境

一、环境配置版本信息:

安装tensorflow-gpu,需特别注意tensorflow-gpu、Python、CUDA、cuDNN版本的适配信息,版本不适配会导致tensorflow-gpu安装失败,该安装教程选择的软件版本信息为:ubuntu18.04 + Anaconda3.5.3.1 + Python3.6.12 + tensorflow-gpu2.2.0 + CUDA10.1 + cuDNN7.6.5

更多版本适配信息可参考官网:https://tensorflow.google.cn/install/source_windows

在这里插入图片描述

二、安装步骤:

1、安装Anaconda:

https://blog.csdn.net/a745233700/article/details/109376667

2、使用Anaconda创建Python环境:

(1)基于python3.6创建一个名为py36的环境

conda create -n py36 python=3.6.12

(2)激活环境:

conda activate py36

3、安装tensorflow-gpu 2.2.0:

pip install tensorflow-gpu==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

4、安装CUDA 10.1

conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

5、安装cuDNN 7.6.5:

conda install cudnn=7.6.5

6、测试是否安装成功:

import tensorflow as tf
# 检查tensorflow是否得到CUDA支持,安装成功则显示true,否则为false
tf.test.is_built_with_cuda()
# 检查tensorflow是否可以获取到GPU,安装成功则显示true,否则为false
tf.test.is_gpu_available()

三、遇到的问题及解决方案:

1、安装完Anaconda后,会有一个默认的base运行环境,能否直接在默认的环境中安装tensorflow?还是必须新建一个新的运行环境?

不要直接使用默认的环境安装,最好不同任务使用不同环境,在默认环境安装新的模块可能会有冲突,导致Anaconda崩溃,最终需要卸载重装。
在默认base运行环境安装tensorflow-gpu 2.2.0的过程中,由于需更新Python的版本,会导致与原本的模块冲突而造成Anaconda崩溃。

2、tf.test.is_gpu_available()的运行结果为false

(1)首先确保tensorflow-gpu、Python、CUDA、cuDNN版本的适配信息,如果不适配,则卸载不适配的版本并重装,如果版本适配,则进入步骤(2);

(2)配置cuda的环境变量:

打开 ~/.bashrc (vim ~/.bashrc),配置下面的环境变量:

export CUDA_HOME=/root/anaconda3/pkgs/cudatoolkit-10.1.243-h6bb024c_0/lib
export PATH=$PATH:$CUDA_HOME/bin

CUDA_HOME指向cuda包的安装路径,anaconda安装的软件包默认都放在/root/anaconda3/pkgs路径下

(3)更新环境变量:

source ~/.bashrc

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/100039.html原文链接:https://javaforall.cn

【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛

【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...

(0)


相关推荐

  • 【STM32】HAL库 STM32CubeMX教程十—DAC「建议收藏」

    【STM32】HAL库 STM32CubeMX教程十—DAC「建议收藏」前言:本系列教程将对应外设原理,HAL库与STM32CubeMX结合在一起讲解,使您可以更快速的学会各个模块的使用所用工具:1、芯片:STM32F407ZET6/STM32F103ZET62、STM32CubeMx软件3、IDE:MDK-Keil软件4、STM32F1xx/STM32F4xxHAL库知识概括:通过本篇博客您将学到:DAC工作原理STM32CubeMX创建…

  • TDA2030的功率放大电路详细教程「建议收藏」

    TDA2030的功率放大电路详细教程「建议收藏」本电路可以将是利用运放TDA2030A制作的功率放大器。电源电压为±12V至±22V。输出的最大功率为18W。该电路为深度负反馈电路,输出电压的放大倍数约为Av=R1/R2=32.3(具体放大倍数请参考模电书籍负反馈部分)。其中R4选用大功率水泥电阻,因为空载时流过R4的电流会过大。D1与D2为二极管,有黑线或者银色线的一端为负极。没有标有正负号的电容为无极电容,不需要区别正负极。标有正负…

  • plsqldev715 注册码「建议收藏」

    plsqldev715 注册码「建议收藏」plsqldev715注册码QVJTSV-KNNSSK-G4UUL4-354LGX-FRZ8X3112243.957xs374ca

  • Java安全之JBoss反序列化漏洞分析

    Java安全之JBoss反序列化漏洞分析0x00前言看到网上的Jboss分析文章较少,从而激发起了兴趣。前段时间一直沉迷于工具开发这块,所以打算将jboss系列反序列化漏洞进行分析并打造成GUI

    2021年12月13日
  • eth挖矿软件_PI挖矿

    eth挖矿软件_PI挖矿Gpu矿机使用(文章测试使用的是AMD580显卡,Ethminer0.17.0)先执行ethminer-G–list-devices查看Gpu数量,我的机器是8Gpu,执行命令后1首先启动geth节点   geth–port33333–rpc–rpcapieth–rpcaddr192.168.10.176–rpcport8888conso…

  • Alex 的 Hadoop 菜鸟教程: 第21课 不只是在HBase中用SQL:Phoenix

    Alex 的 Hadoop 菜鸟教程: 第21课 不只是在HBase中用SQL:Phoenix什么是Phoenix?Phoenix的团队用了一句话概括Phoenix:”WeputtheSQLbackinNoSQL”意思是:我们把SQL又放回NoSQL去了!这边说的NoSQL专指HBase,意思是可以用SQL语句来查询Hbase,你可能会说:“Hive和Impala也可以啊!”。但是Hive和Impala还可以查询文本文件,Phoenix的特点就是,它只能查Hbase,别的类型都不支持!但是也因为这种专一的态度,让Phoenix在Hbase上查询的性能超过了Hive和Impala!

发表回复

您的电子邮箱地址不会被公开。

关注全栈程序员社区公众号