rocm平台_利用ROCm在3A平台进行深度学习

零、平台介绍

0、软件平台

操作系统:Ubuntu20.04

Linux Kernel:5.8.4-050804-generic

NVCC(ROCm):rocm-dev3.7.0/Ubuntu 16.04 3.7.0-20 amd64

Docker:docker-ce/focal,now 5:19.03.12~3-0~ubuntu-focal amd64

1、硬件平台

CPU:Ryzen 3700X

GPU:Radeon VII

主板:ROG STRIXB450-I

内存:3200MHz 2*16GB

2、ROCm介绍

ROCm是AMD显卡的开放计算平台,提供了接近Cuda的API和转换工具。目前支持Pytorch和TensorFlow。

一、平台搭建

0、安装ROCm

(1)更新到最新版本内核

前往 https://kernel.ubuntu.com/~kernel-ppa/mainline/v5.8.4/ 下载内核文件 :

linux-headers-5.8.4-050804-generic_5.8.4-050804.202008260637_amd64.deb

inux-headers-5.8.4-050804_5.8.4-050804.202008260637_all.deb

linux-image-unsigned-5.8.4-050804-generic_5.8.4-050804.202008260637_amd64.deb

linux-modules-5.8.4-050804-generic_5.8.4-050804.202008260637_amd64.deb

切换到文件目

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值