最新避坑Ubuntu20.04安装tensorflow-gpu

本文详细介绍了在Ubuntu20.04系统上安装tensorflow-gpu2.6.0的全过程,包括查询匹配配置、安装Anaconda3、下载安装cuda和cudnn,以及配置Pycharm和测试GPU功能。关键步骤包括使用Anaconda3的特定版本,避免直接从Nvidia官网下载,以及激活虚拟环境进行安装。
摘要由CSDN通过智能技术生成

最新!避坑!Ubuntu20.04安装tensorflow-gpu2.6.0有效!

1.基本信息&查询对应配置:

系统:Ubuntu20.04
显卡:RTX3090 ×2
python版本:3.7
安装的TF-gpu版本:2.6.0
官网看一下tf,python,cuda,cudnn的匹配,我是装的下图那一行,没啥,就是不敢用最新的。
我选择的我选中的红色那一行对应安装的

2.安装Anaconda3

去这个地址下载早一些版本的Anaconda,我下载的是Anaconda3-5.3.1-Linux-x86_64.sh这个版本。
cd到下载目录,然后安装,期间一直yes,回车就行了。

$bash Anaconda3-5.3.1-Linux-x86_64.sh

接下来将anaconda中自带的python3.7添加到环境,用以区分系统自带的python

cd ~
sudo vim .bashrc

将下面两行添加到末尾

  • 8
    点赞
  • 30
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
要在Ubuntu 20.04安装TensorFlow-GPU,您需要先安装CUDA和cuDNN。根据引用\[2\]中提供的信息,您可以按照以下步骤进行安装: 1. 首先,您需要下载CUDA Toolkit 11.2。您可以使用引用\[1\]中提供的链接下载CUDA Toolkit 11.2的安装程序。 2. 下载完成后,打开终端并导航到下载的安装程序所在的目录。运行以下命令以安装CUDA Toolkit 11.2: ``` sudo sh cuda_11.2.0_460.27.04_linux.run ``` 3. 安装过程中,您需要按照提示进行操作。请注意,安装过程可能需要一些时间。 4. 安装完成后,您需要安装cuDNN。您可以在NVIDIA的官方网站上下载适用于CUDA 11.2的cuDNN库。 5. 下载完成后,解压缩cuDNN文件并将其复制到CUDA的安装目录。具体的复制命令可以在cuDNN的安装文档中找到。 6. 安装完成后,您可以继续安装TensorFlow-GPU。您可以使用pip命令来安装TensorFlow-GPU: ``` pip install tensorflow-gpu ``` 7. 安装完成后,您可以验证TensorFlow-GPU是否正确安装。在终端中运行以下命令: ``` python ``` 然后在Python交互式环境中输入以下代码: ``` import tensorflow as tf print(tf.__version__) print(tf.test.is_gpu_available()) ``` 如果输出显示您安装TensorFlow-GPU版本和GPU可用性为True,则说明安装成功。 请注意,这只是一个大致的安装过程,具体步骤可能会因您的系统配置和需求而有所不同。建议您在安装之前参考引用\[2\]中提供的博客文章以获取更详细的安装指南。 #### 引用[.reference_title] - *1* *2* [Ubuntu 20.04 安装 tensorflow-gpu](https://blog.csdn.net/weixin_44709392/article/details/123800999)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [ubuntu20.04 搭建深度学习环境tensorflow-gpu2.*](https://blog.csdn.net/qq_43288602/article/details/125153866)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值