Linux服务器安装Kokkos-core 和 Kokkos-kernel
说明
由于实验室项目原因,需要跑一个Gmres算法,之前弄过kokkos,就想在kokkos-kernels里跑现有的GMRES算法库
在此记录自己的安装的过程,以及自己踩过的一些坑。
1. 准备工作
从Kokkos官网下载Kokkos 以及 Kokkos-kernels:
https://github.com/kokkos/kokkos.git -- Kokkos-core
https://github.com/kokkos/kokkos-kernels.git -- Kokkos-kernels
注意: 如果是压缩包下载到服务器上的话,编译Kokkos-core的时候需要先将Kokkos-master/bin/
里面的四个脚本加上可执行权限,不然编译的时候会报错。
2. 安装Kokkos-core
然后在kokkos-master/
目录下
mkdir build && cd build cmake .. // 若要指定CUDA后端则加上编译选项 -DKokkos_ENABLE_CUDA=ON -DKokkos_ARCH_PASCAL61=ON(可在https://kokkos.github.io/kokkos-core-wiki/keywords.html找到相应的编译选项 ) make -j make install // 可能要加上sudo权限
make install
的时候可以看下路径,后面编译kokkos-kernel的时候需要指定Kokkos库的安装路径
至此Kokkos-core就已经安装好了
3. 安装Kokkos-kernel
在目录kokkos-kernels-master
目录下
mkdir build && cd build cmake .. -DCMAKE_BUILD_TYPE=Release -DKokkosKernels_ENABLE_TESTS=ON -DKokkosKernels_ENABLE_EXAMPLES=ON -DKokkos_ROOT=/usr/local -DKokkos_ENABLE_CUDA=ON // 此处我指定后端为cuda并且指定相应的架构 make -j
然后在 build
目录下,example
中就有相关test代码啦,我此次测试的是gmres就在build/example/gmres
下面。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!