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库的安装路径
image
至此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下面。
image

posted @ 2023-05-26 12:03  人生逆旅,我亦行人  阅读(273)  评论(0编辑  收藏  举报