6700xt+Rocm5.4使用Tensorflow与Pytorch
环境:Opensuse Leap15.4(如果你是风吹草用户下面有额外描述
#参考https://www.videogames.ai/2022/09/01/RX-6700s-Machine-Learning-ROCm.html
安装这个研究了很久,制作不宜,需要转载请标注来源
[修改清华源(可以不改)]
禁用官方软件源sudo zypper mr -da添加 TUNA 镜像源sudo zypper ar -cfg 'https://mirrors.tuna.tsinghua.edu.cn/opensuse/distribution/leap/$releasever/repo/oss/' tuna-oss
sudo zypper ar -cfg 'https://mirrors.tuna.tsinghua.edu.cn/opensuse/distribution/leap/$releasever/repo/non-oss/' tuna-non-oss
sudo zypper ar -cfg 'https://mirrors.tuna.tsinghua.edu.cn/opensuse/update/leap/$releasever/oss/' tuna-update
sudo zypper ar -cfg 'https://mirrors.tuna.tsinghua.edu.cn/opensuse/update/leap/$releasever/non-oss/' tuna-update-non-ossLeap 15.3 用户还需添加 sle 和 backports 源sudo zypper ar -cfg 'https://mirrors.tuna.tsinghua.edu.cn/opensuse/update/leap/$releasever/sle/' tuna-sle-update
sudo zypper ar -cfg 'https://mirrors.tuna.tsinghua.edu.cn/opensuse/update/leap/$releasever/backports/' tuna-backports-update
[安装AMDGPU]
#第一部,安装AMDGPU
sudo zypper --no-gpg-checks install
https://repo.radeon.com/amdgpu-install/5.4/sle/15.4/amdgpu-install-5.4.50400-1.noarch.rpm
#安装(注意这里会提示缺少依赖)
sudo amdgpu-install --usecase=rocm,hip,mllib --no-dkms
#安装缺少的依赖(注意一定要直接抓取二进制安装(那个源有问题))
Install package devel:languages:perl / perl-URI-Encode
#添加密钥选择信任就行了
#添加环境
sudo usermod -a -G video $LOGNAME
sudo usermod -a -G render $LOGNAME
#在文件执行
vim ~/.bashrc #添加下面
export HSA_OVERRIDE_GFX_VERSION=10.3.0 #如果你是amd显卡6700xt
export LD_LIBRARY_PATH=/opt/rocm/lib
source ~/.bashrc #保存文件
#复制一下链接文件
cd $LD_LIBRARY_PATH
cp librocblas.so.0 librocblas.so
cp libMIOpen.so.1 libMIOpen.so
cp librocrand.so.1 librocrand.so
#安装gcc和g++(opensuse默认不带),以及miopen内核
sudo zypper install rccl
sudo zypper install gcc
sudo zypper install gcc-c++
sudo zypper install miopenkernels-gfx1030-36kdb5.4.0
#安装anaconda
Anaconda | The World's Most Popular Data Science Platform #在官网下载
chmod +x <下载的软件>.sh #添加执行权限
./<下载的软件>.sh #执行
#更新anaconda
conda update --all
#安装tensorflow
conda create -n mytensorflow python=3.8 #我使用的3.8版本,你也可以试别的版本
conda activate mytensorflow
pip install tensorflow-rocm
#安装pytorch
conda create -n mypytorch python=3.9
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2/
#测试tensorflow
git clone GitHub - tensorflow/benchmarks: A benchmark framework for Tensorflow
cd benchmarks/scripts/tf_cnn_benchmarks
python3 tf_cnn_benchmarks.py --num_gpus=1 --batch_size=32 --model=resnet50
6700xt+Rocm5.4使用Tensorflow与Pytorch相关推荐
- TensorFlow与PyTorch模型部署性能比较
TensorFlow与PyTorch模型部署性能比较 前言 2022了,选 PyTorch 还是 TensorFlow?之前有一种说法:TensorFlow 适合业界,PyTorch 适合学界.这种说 ...
- 深度学习调用TensorFlow、PyTorch等框架
深度学习调用TensorFlow.PyTorch等框架 一.开发目标目标 提供统一接口的库,它可以从C++和Python中的多个框架中运行深度学习模型.欧米诺使研究人员能够在自己选择的框架内轻松建立模 ...
- 常用深度学习框——Caffe/TensorFlow / Keras/ PyTorch/MXNet
常用深度学习框--Caffe/TensorFlow / Keras/ PyTorch/MXNet 一.概述 近几年来,深度学习的研究和应用的热潮持续高涨,各种开源深度学习框架层出不穷,包括Tensor ...
- C++调用Python文件,TensorFlow和PyTorch构建的深度学习模型,无法使用GPU的情况分析。
C++调用Python深度学习模型,包含TensorFlow和PyTorch等构造的模型,然后使用GPU出现问题.包含C++调用Python函数,C++加载模型到GPU,GPU内存占用过大,计算完毕内 ...
- GitHub:TensorFlow、PyTorch最全资料集锦
给各位小伙伴们推出几个深度学习框架的资料集锦,统一命名为:XXX-From-Zero-To-One.下面po一幅深度学习框架发展的重要历史点: 从上图可知,TensorFlow和PyTorch是目前深 ...
- TensorFlow、PyTorch夹缝之下:后浪的进击和野望
作者 | Just 出品 | AI科技大本营(rgznai100) "我发现,软件研发总会延期."一流科技CEO袁进辉说. 按照他的预期,深度学习框架OneFlow做两年就能开源给 ...
- TensorFlow、PyTorch之后,“国产”AI框架还有没有机会?
出品 | AI科技大本营(ID:rgznai100) 众所周知,在机器学习框架领域,PyTorch.TensorFlow已分别成为目前学界和业界使用最广泛的两大实力玩家,而紧随其后的Keras.MXN ...
- 2020,国产AI开源框架“亮剑”TensorFlow、PyTorch
「AI技术生态论」 人物访谈栏目是CSDN发起的百万人学AI倡议下的重要组成部分.通过对AI生态专家.创业者.行业KOL的访谈,反映其对于行业的思考.未来趋势的判断.技术的实践,以及成长的经历. 20 ...
- 又是华为!对标 TensorFlow、PyTorch,深度学习框架 MindSpore已开源!附入手公开课...
Datawhale 作者:Datawhale 摘要:随着深度学习框架等工具的全面开源,最强 AI 处理芯片昇腾的落地,华为的 AI 布局已经逐渐清晰起来,MindSpore 将为各行业带来哪些新变化, ...
最新文章
- 京东某女程序员求助:刚入职就意外怀孕,纠结还能不能过试用期?网友:职场女性太难!...
- R语言可视化包ggplot2改变图例(legend)元素的大小实战:包含图例中标题字体、文本字体、标识模块(key)的大小
- mask - 使用 * 遮蔽字符串
- hdu4848 DFS 暴搜+ 强剪枝
- linux redhat、ubuntu系统 docker启动、停止命令
- 数据库:MySQL 团队开发规范,太详细了,建议收藏!
- CreateThread和_beginthreadex的区别
- shell循环,判断介绍,以及实例
- 面试官:HashMap 为什么线程不安全?
- python鼠标右键检测按钮_如何检测右键单击+左键单击
- 51nod1432 独木舟
- sql重命名数据库_SQL重命名数据库
- java 二叉树排序算法
- 微信小程序 使用cookie
- 读书笔记:《重来REWORK》
- 配置Java远程监控授权(Java Mission Control)
- 北风网盘点2016年中国程序员薪酬状况
- 从关山口到五道口(2019年清华计算机考研全程回顾+经验+总结)
- dash live 播放
- iOS -- 友盟工具进行Crash分析/dsym文件