python调用gpu amd_TensorFlow使用AMD GPU实现加速（ROCm/Ubuntu 18.04）

最新推荐文章于 2024-08-31 07:11:24 发布
最新推荐文章于 2024-08-31 07:11:24 发布
阅读量1.5k 收藏 7
点赞数
文章标签： python调用gpu amd
本文介绍了如何在Ubuntu 18.04系统中，利用AMD的ROCm平台配置AMD Radeon RX580 GPU以加速TensorFlow。详细步骤包括安装AMD GPU驱动、ROCm组件以及TensorFlow的ROCm版本，最后验证了GPU加速成功。
摘要由CSDN通过智能技术生成

本文通过AMD开发ROCm平台，Ubuntu 18.04系统中，TensorFlow也能使用AMD GPU实现GPU加速。现在把具体搭建流程呈上。

电脑硬件信息

CPU：AMD Ryzen 1700x

GPU：AMD Radeon RX580

内存：32G

硬盘：SSD 256GB + HDD 2TB

系统安装的是Ubuntu 18.04

本文针对的是Ubuntu 18.04，我选的是最小安装方式，当然也可以安装优麒麟Ubuntu Kylin 18.04，可以参考教程：https://ywnz.com/linuxaz/1443.html

安装AMD GPU驱动程序

1.下载最新的驱动程序，我使用的是18.20版本。

里面包含for RHEL 7.5/CentOS 7.5、RHEL 6.9/CentOS 6.9、Ubuntu 18.04、Ubuntu 16.04.4、SLED/SLES 12 SP3版本，选择Radeon Software for Linux version 18.20 for Ubuntu 18.04版本下载。

2.以下载到Downloads目录为例：

cd ~/Downloads

tar -Jxvf amdgpu-pro-18.20-606296.tar.xz

cd ~/Downloads/amdgpu-pro-18.20-606296

./amdgpu-pro-install --opencl=legacy

安装ROCm

1.增加ROCm的仓库

wget -qO - http

最低0.47元/天 解锁文章
weixin_39783633
关注
程序员 笔记本 amd python_如何在AMD GPU上运行Python？
weixin_39932300的博客
11-30 1243
如何在AMD GPU上运行Python？
python调用gpu amd_TensorFlow通过AMD GPU加速（ROCm/elementary OS 5.0）
weixin_36205186的博客
02-04 1592
elementary OS 5.0是基于Ubuntu 18.04制作，所以本流程也适用于Ubuntu 18.04。ROCm升级1.91之后不需要安装AMD的GPU驱动程序，如果安装可能会出现问题。安装elementary OS 5 之后的一些配置安装软件更新管理器sudo apt updatesudo apt install software-properties-common software-...
加速 PyTorch 模型：使用 ROCm 在 AMD GPU 上应用 torch.compile
最新发布
数字人生
08-31 2051
PyTorch 2.0 引入了一个名为**的工具，可以极大地加速 PyTorch 代码和模型。通过将 PyTorch 代码转换为高度优化的内核，`torch.compile` 在现有代码库上进行最小化修改即可提供显著的性能提升。此功能允许精确优化单个函数、整个模块以及复杂的训练循环，提供了一个多功能且强大的工具来提高计算效率。在这篇博客中，我们将演示如何在 AMD GPU 上使用 ROCm 和加速各种实际模型。
tensorflow使用gpu加速
Wicher_D的博客
04-24 3063
所以，这里需要根据我们前面的python版本确定对应的CUDA和cuDNN，这里我们使用了python3.9，所以确定对应的CUDA为11.2、cuDNN为8.1。这里需要注意的是，tensorflow是我自定义的环境名，可定义为你喜欢的不带空格和中文的其它名字，python版本要安装tensorflow对应的，可参考下面的网址来确定，这里我以python3.9为例。这里需要注意的是，tensorflow库在2.12及以上的版本中，都取消了对Windows平台的GPU支持（，根据你显卡的版本来下载。
配置Tensorflow使用CUDA进行GPU加速(超详细教程)
热门推荐
m0_51302496的博客
04-01 2万+
对于刚使用Tensorflow的友友来说配置环境并使用GPU进行加速也是件令人头疼的事情，纯自己折腾会遇到比较多的坑，所以这里详细介绍一下Tensorflow的环境配置
配置Tensorflow使用GPU加速（Tensorflow2.6.0，Python3.6-3.9，CUDA11.2，cuDNN8.1）
m0_68167270的博客
06-04 3607
对于刚使用Tensorflow的朋友来说，配置环境并使用GPU进行加速也是件令人头疼的事情，纯自己折腾会遇到比较多的坑，我对我安装Tensorflow的过程进行总结，我所安装的Tensorflow版本是2.6.0，与之对应的Python为3.6-3.9，CUDA为11.2，cuDNN为8.1。
rocm平台_TensorFlow通过AMD GPU加速（ROCm/Ubuntu 18.04）
weixin_29442195的博客
01-17 2033
ROCm 1.91之后不需要安装AMD GPU驱动程序。请参考新的安装流程：通过AMD开发ROCm平台，TensorFlow可以使用AMD GPU实现GPU加速。现将搭建流程呈上。硬件：CPU：AMD Ryzen 1700xGPU：AMD Radeon RX580内存：32G硬盘：SSD 256GB + HDD 2TB安装Ubuntu 18.04网上很多Ubuntu安装教程，这里不在赘述。我选的是...
python调用gpu amd_如何在AMD的GPU上运行TensorFlow？
weixin_39640085的博客
12-09 2978
工资不涨，英伟达 GPU 的售价年年涨。因此，多一个竞争对手，总是好事。近日，Google 宣布推出适用于 ROCm GPU 的 TensorFlow v1.8，其中包括 Radeon Instinct MI25。对于 AMD 正在进行的深度学习加速工作而言，这是一座重大的里程碑。ROCm即 Radeon 开放生态系统 (Radeon Open Ecosystem)，是我们在 Linux 上进行...
pytorch 1.4 on ROCM 3.1 （AMD GPU) for ubuntu 18.04
03-13
AMD GPU用户的福音。用AMD GPU学习人工智能吧。...pytorch 1.4 on ROCM 3.1 for ubuntu 18.04, ROCM 版本 ３．１ 预编译版本，直接pip install xxxx.whl 使用，省事省力。提前安装好rocm平台驱动。
Mac 安装 TensorFlow 2.7 环境，支持 AMD 显卡
一条老萌新
01-07 2578
先抛出问题，望高人指点迷津： 跑 mnist 的 demo，GPU 比 CPU 慢了很多，为啥呢？ 本机环境： CPU：Intel I9 8核16线程 内存：64G 显卡：AMD Radeon Pro 5500M 示例代码 import tensorflow as tf def run(): mnist = tf.keras.datasets.mnist (x_train, y_train), (x_test, y_test) = mnist.load_data()
Ubuntu配置深度学习环境（TensorFlow和PyTorch）
zardforever123的博客
09-30 8683
前言：最好是去[这里](https://tensorflow.google.cn/install/source?hl=en#linux)看一下TensorFlow对应的CUDA版本，按照要求的版本安装，我是因为有其他程序必须要11.6才没有按照这个要求安装。
python调用gpu amd,在AMD GPU中将Python + Theano与OpenCL一起使用
weixin_34614567的博客
12-09 371
I'm trying to use Python with Theano to accelerate some code with OpenCL. I installed libgpuarray and pygpu as instructed (I think), and got no errors. The installation detected the OpenCL runtime ins...
ROCm平台搭建tensorflow2.0 踩坑日记
qq_37650969的博客
11-17 1477
tensorflow-addon在我写在这篇博文时还未被支持 RNN style的layer请使用tf.compat.v1.keras.layers.LSTM而不是tf.keras.layers.LSTM，因为后者默认开启CuDNN加速，AMD显卡显然是不支持的 ...
Linux安装Tensorflow2
sievert的博客
06-16 1万+
假如你不幸和曾经的我一样，需要将TF2安装在非官撑的其他平台，那你还有两种方法可以去尝试，一种是借助docker，将环境部署在容器里；如果你需要更多更高级的配置，完全可以通过Dockerfile以及docker compose来实现，这里就不赘述了。章节三中的源代码构建方式更多的用途是基于TF2源码进行解析及二次开发，而通过docker方式进行部署或测试要比打包成PIP包的形式来的更加方便。假如你使用包管理器，直接创建一个环境即可。更不用提什么WSL，开发板，统统是不关心，不支持，真有你的Google。
CentOS 7.5 安装GPU环境（显卡驱动，CUDA，CuDNN）以及 Python3.6
学无止境
03-14 1392
文章目录CentOS解决lspci(command not found) 问题安装GPU环境1. 安装显卡驱动2. 安装CUDA（1）检查环境（2）安装CUDA3. 安装cuDNN4. 安装python3.6.45.虚拟环境安装和TensorFlow-GPU安装Centos一些常用命令参考文章 CentOS解决lspci(command not found) 问题 参考：https://blog....
推荐一个apt命令加速神器
编程那些年的博客
01-17 947
推荐一个apt命令加速神器 有时候使用apt/apt-get命令时，会遇到下载比较慢的情况，特别是遇到有需要安装PPA上的软件的时候，这时就可以用apt-fast神器来对其进行加速。其实现原理是利用了aria2进行多线程下载，原理这里不做展开。仅讲解下如何在Ubuntu和在树莓派中安装该神器。 测试环境 基础系统：Ubuntu 18.04 树莓派系统：Debian armhf版本 Ubuntu上的安装方法 首先、添加apt-fast的安装源： linux@ubuntu:~$sudoadd..
ROCm与tensorflow在ubuntu 18.04下的安装教程
Pointer的博客
06-17 1万+
ROCm与tensorflow在ubuntu 18.04下的安装教程     本来准备买Nvidai的显卡，可是囊中羞涩再加上听说AMD的卡也能使用tensorflow进行深度学习加速了，所以1149京东买了个蓝宝石RX580 8G 白金版。网上找安装教程时发现教程非常少，所以决定写一篇详细的安装攻略，以便下次重装系统时少走弯路。 一、计算机配置环境 CPU：  intel i7 8700 GPU...
windows配置wsl，Unbuntu启动GPU加速
u014196765的博客
08-28 1744
wsl全称Windows Subsystem for Linux，windows电脑下的linux子系统，对于想用Linux的Windows用户来说wsl是一个不错的选择。
ubuntu22.04 amd显卡 安装rocm教程
qq_39103891的博客
03-27 1519
筛选 torch 版本为上面安装的版本： torch2.2.2 对应 torch-2.2.2。如果出错一般是下载的whl 的 文件跟 python 和 rocm torch 版本对不上。筛选 rocm 版本为上面安装的版本： rocm5.7 对应 rocm5.7。筛选 python 版本为上面安装的版本： python3.11 对应 cp311。当前 rocm 版本为 5.7, torch 版本为 2.2.2，记住当前版本号。返回 true 即可。
ubuntu 配置keras_使用AMD GPU和Keras训练神经网络
05-31
5. 配置Keras和TensorFlow以使用AMD GPU：在Python脚本中添加以下代码来配置Keras和TensorFlow以使用AMD GPU： ``` import tensorflow as tf import keras.backend.tensorflow_backend as KTF config = tf....
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值