零、平台介绍
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
切换到文件目