首页 > 其他分享 >CK-Composable Kernel 下载安装 (AMD-Rocm环境)

CK-Composable Kernel 下载安装 (AMD-Rocm环境)

时间:2024-08-09 11:38:57浏览次数:16  
标签:CK Kernel 架构 make AMD build 下载安装 GPU

CK-Composable Kernel 下载安装

AMD 可组合内核库:只需几行代码即可为 AI 应用提供高效的融合内核

CK 库是为 AMD-GPU 所开发的,下载和使用都需要在 AMD-Rocm 环境/平台,如果你的 GPU 并非 AMD 的,是无法下载使用 CK 的。

关于CK库

下载

从 GitHub 存储库下载 CK 源代码并开始构建。 将会得到一个名为composable_kernel的文件夹。

git clone https://github.com/ROCm/composable_kernel.git
cd composable_kernel

安装 (Python + Cmake)

安装Python依赖
cd docs
pip3 install -r sphinx/requirements.txt
python3 -m sphinx -T -E -b html -d _build/doctrees -D language=en . _build/html
查看GPU架构

CK支持的GPU架构有 gfx908;gfx90a;gfx940;gfx941;gfx942;gfx1030;gfx1100;gfx1101;gfx1102

需要查看自己本地(或你的服务器)使用的架构,以下两行代码都可以帮助你明确你所使用的架构。

rocminfo
rocm-smi

我的GPU架构是 gfx942 。因此在下面的cmake配置中,我的 GPU_TARGETS 项为 gfx942

使用Cmake安装

创建构建目录并进入。

mkdir build
cd build

使用以下命令配置cmake, 注意路径,并把 GPU_TARGETS 项更改为你的GPU架构。

cmake \
-D CMAKE_PREFIX_PATH=/opt/rocm \
-D CMAKE_CXX_COMPILER=/opt/rocm/bin/hipcc \
-D CMAKE_BUILD_TYPE=Release \
-D GPU_TARGETS="gfx942"   \
..

构建CK库 并安装CK

make -j
make -j install
示例测试

生成示例和测试

make -j examples tests

生成并运行所有示例和测试

make -j check

标签:CK,Kernel,架构,make,AMD,build,下载安装,GPU
From: https://www.cnblogs.com/beiwei31/p/18350503

相关文章

  • docker搭建本地仓库
    环境准备:服务器:9.134.130.35私有仓库服务器,运行registry容器客户端:9.208.244.175测试客户端,用于上传、下载镜像文件测试搭建本地仓库mkdir/docker/registry-p--创建主机目录dockerrun-itd-v/docker/registry/:/docker/registry-p5000:5000--restart=always--......
  • Local All-Pair Correspondence for Point Tracking 中英对照
    论文来自:https://ku-cvlab.github.io/locotrack/LocalAll-PairCorrespondenceforPointTracking局部全对应对点跟踪SeokjuCho\({}^{1}\),JiahuiHuang\({}^{2}\),JisuNam\({}^{1}\),HonggyuAn\({}^{1}\),Seungryong\({\mathrm{{Kim}}}^{1,\dagger}\),......
  • Docker Desktop 出现 Unable to calculate image disk size 错误提示的解决方法
    Win11安装了DockerDestop,版本为4.25.2。DockerDesktop能随系统启动而启动,其内部管理的容器也可以正常启动,但是不和谐的地方也出现了:在从通知区域双击DockerDesktop的图标后,它就会出现如下错误信息,虽不影响使用,但是相当碍眼:  解决方案:【1】进入设置,关闭“UsetheWSL2......
  • docker pull实现断点续传
    问题背景在使用Docker拉取DockerHub的镜像时,经常会出现网络不稳定的问题,这就导致拉取到一半的镜像会重新拉取,浪费时间。例如下面这种情况:第二次拉取这是一个网络中断的场景,第二次重新拉取的时候,同样是d103这个部分,又重新下载了一次。而且在不稳定的网络情况下(大概率),有可......
  • 快速基于 ClickHouse + Grafana 搭建可观测性解决方案 - 分布式链路追踪篇(ClickHouse
    引言在ClickHouse,我们认为可观测性仅仅是另一个实时分析问题。作为一款高性能的实时分析数据库,ClickHouse被用于多种场景,包括时间序列数据的实时分析。其应用场景的多样性推动了大量分析函数的发展,这些函数有助于查询大多数数据类型。这些查询特性和高压缩率使得越来越多的用户......
  • docker 构建redis
    一、Docker搜索redis镜像dockersearchredis 二、Docker拉取镜像dockerpullredis 三、Docker挂载配置文件接下来就是要将redis的配置文件进行挂载,以配置文件方式启动redis容器。(挂载:即将宿主的文件和容器内部目录相关联,相互绑定,在宿主机内修改文件的话也随之修改容......
  • dockker部署redis报错(报:WARNING Memory overcommit must be enabled! Without it, a b
    转载博客(机翻):https://ourcodeworld.com/articles/read/2083/how-to-remove-redis-warning-on-docker-memory-overcommit-must-be-enabledHowtoremoveRediswarningonDocker:Memoryovercommitmustbeenabled如何删除Docker上的Redis警告:必须启用内存过量使用Carlos......
  • asp.net core 6 发布docker
    一、在项目当中创建dockerfile文件设置dockerfile属性为“始终复制”,FROMmcr.microsoft.com/dotnet/aspnet:6.0ASbaseWORKDIR/webCOPY..EXPOSE80EXPOSE443FROMbaseASfinalENTRYPOINT["dotnet","OneZhanMVC.dll"]二、创建镜像cd项目目录......
  • Memcheck错误解释
    Memcheck只能检测出两种错误类型:useofillegaladdressesuseofundefinedvalues这两种错误类型,足以应对大多数的情况。下面针对这两种类型出现的错误进行解释。Illegalread/IllegalwriteerrorsInvalidreadofsize1at0x4C32CF2:strlen(in/usr/li......
  • 构建即时通讯应用:Spring boot高效集成WebSocket、Stomp协议完成前后端持续通信
    1.引入依赖在你的SpringBoot项目的pom.xml中添加以下依赖:<dependencies><!--SpringBootStarterThymeleaf--><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-st......