安装
这里我们展示如何在 Ubuntu 24.04 和最新的 NVIDIA 产品上安装 gRASPA
先决条件
- 一台带有 NVIDIA GPU 的机器
- Ubuntu 24.04(最好是干净版本,但不是必须)
步骤(如果你从干净的 Ubuntu 24.04 开始)
1. 安装 NVIDIA 驱动
- 找到推荐的 NVIDIA 驱动
2. 安装 NVIDIA CUDA工具包
- 打开 终端
- 在终端输入:
sudo apt install nvidia-cuda-toolkit
3. 安装 NVHPC编译器
- 访问最新的 NVHPC 编译器网站
- 最新版本是24.5
- 选择下载方法(例如tar文件)
- 跟着他们的说明
- 注意查看安装程序打印到屏幕上的最终消息
- 你会注意到一行添加
nvhpc
到PATH
:export PATH=/opt/nvidia/hpc_sdk/Linux_x86_64/24.5/compilers/bin:$PATH
- 复制这行! 以便稍后使用
4. 下载 gRASPA
cd ~/
git clone https://github.com/snurr-group/gRASPA
5. 编译 gRASPA
cd gRASPA/src_clean
cp ../NVC_COMPILE .
- 打开 NVC_COMPILE
- 在 NVC_COMPILE 顶部添加(你刚刚复制的那行)
export PATH=/opt/nvidia/hpc_sdk/Linux_x86_64/24.5/compilers/bin:$PATH
- 在 NVCFLAG 中添加
-target=gpu
(更简单的选项)- 如果使用
-target=gpu
,移除-gpu=ccXX
- 如果使用
-gpu=ccXX
,你需要确定你的 GPU 的ccXX
- 如果使用
chmod 777 NVC_COMPILE
./NVC_COMPILE
你的编译文件可能看起来像这样:
#!/bin/bash rm nvc_main.x export PATH=/opt/nvidia/hpc_sdk/Linux_x86_64/24.5/compilers/bin:$PATH CXX="nvc++" LINKFLAG="-L/usr/lib64/ -L/opt/local/lib/gcc11/" NVCFLAG="-O3 -std=c++20 -target=gpu -Minline -fopenmp -cuda -stdpar=multicore" $CXX $NVCFLAG $LINKFLAG -c axpy.cu $CXX $NVCFLAG $LINKFLAG -c main.cpp $CXX $NVCFLAG $LINKFLAG -c read_data.cpp $CXX $NVCFLAG $LINKFLAG -c data_struct.cpp $CXX $NVCFLAG $LINKFLAG -c VDW_Coulomb.cu $CXX $NVCFLAG $LINKFLAG main.o read_data.o axpy.o data_struct.o VDW_Coulomb.o -o nvc_main.x rm *.o
6. 运行一个示例(MFI沸石中的二氧化碳吸附)
cd ../Examples/CO2-MFI/
../../src_clean/nvc_main.x