Linux conda tensorflow-gpu安装及Not creating XLA devices, tf_xla_enable_xla_devices not set相关问题解决

0、首先介绍一下conda创建虚拟环境的基本操作。
(1)创建Python的虚拟环境,并指定Python版本,不指定的话会使用默认的版本(Conda Base 环境中的Python版本)。

conda create -n grape python=3.8.8   # grape可以指定为自己想要的名字

(2)进入创建好的虚拟环境。

conda activate grape

(3)退出当前Anaconda虚拟环境,在当前环境中执行以下指令。

conda deactivate

(4)删除这个虚拟环境

conda remove -n grape --all

出现Not creating XLA devices, tf_xla_enable_xla_devices not set问题,一般是由于tensorflow和cuda、cuddn的版本问题。建议直接花几分钟重新配置环境,以下有两种方法安装tensorflow。

1、直接用conda create一个相应版本的tensorflow-gpu虚拟环境。

conda create -n tf tensorflow-gpu==2.2.0

在安装过程中会自己安装相应的cuda、cuddn包。
在这里插入图片描述

2、安装需要的cuda和cudnn环境,再安装tensorflow-gpu。
首先我们需要确定我们需要的tensorflow-gpu的版本,再在tensorflow找到对应的cuda和cudnn版本。网上部分帖子里面的不太对,还是自己去官网看比较靠谱。
https://tensorflow.google.cn/install/source#install_the_package
![在这里插入图片描述](https://img-blog.csdnimg.cn/31aa273563b841419df4cdb9f50ec7d8.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBAYWxwc2lsb24=,size_20,color_FFFFFF,t_70,g_se,x_16
以tensorflow-gpu-2.2.0为例。
(1)首先创建一个python虚拟环境并激活

conda create -n grape python=3.8.8
conda activate grape

(2)在环境中安装cuda

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

(3)在环境中安装cudnn

conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

(4)安装tensorflow-gpu-2.2.0

conda install tensorflow-gpu-2.2.0

有问题欢迎留言交流。

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值