前提是已经安装过wsl
1 在 WSL 2 中启用 NVIDIA CUDA
参考在 WSL 2 上启用 NVIDIA CUDA | Microsoft Learn
注意:勿在 WSL 中安装任何 Linux 显示驱动程序。Windows 显示驱动程序将同时安装本机 Windows 和 WSL 支持的常规驱动程序组件。
2 在WSL2中配置Cuda
不安装任何Linux显示驱动程序,必要的Cuda还是要安装的,以下为具体步骤:
2.1 首先要知道使用的CUDA版本
先打开Windows界面的终端,输入命令查看显卡版本。找到对应的CUDA安装版本
nvidia-smi
如图我的适配CUDA版本最高是12.6,注意此时千万要查看一下AMBER的GPU加速章节,里面会讲解该版本AMBER适配的CUDA版本。
可以看到,我的AMBER24版本支持的CUDA版本最高12.4,而一开始我下载的CUDA版本是12.5,导致我重新安装了一遍。
2.2 前往Nvidia官网下载对应的cuda包,用什么版本就下什么版本
具体怎么下载的参照:WSL——卸载、安装CUDA_wsl2卸载cuda-CSDN博客
3 编译并安装AMBER24 CUDA-串行版
Amber从20开始采用cmake编译。在解压获得的amber24_src的build文件夹中,有一个run_cmake文件。里面是预写好的cmake编译指令,用户可以根据安装需求进行更改。
该文件真正有用的是这几行:
# Assume this is Linux:
cmake $AMBER_PREFIX/amber22_src \
-DCMAKE_INSTALL_PREFIX=$AMBER_PREFIX/amber22 \
-DCOMPILER=GNU \
-DMPI=FALSE -DCUDA=FALSE -DINSTALL_TESTS=TRUE \
-DDOWNLOAD_MINICONDA=TRUE \
2>&1 | tee cmake.log
根据AMBER手册中的步骤,修改run_cmake文件即可,这样的好处是一些软件可以不用重复安装。也说明可以直接安装串行-cuda版,不需要在此之前先安装好串行-cpu版。
根据AMBER手册,只需要对run_cmake文件进行修改 :
-DCUDA=FALSE =》》 -DCUDA=TRUE
然后执行 sudo ./run_cmake ; sudo make install,随后进行测试:
cd $AMBERHOME
export CUDA_VISIBLE_DEVICES=0 (choose the GPU id you wish to test)
make test.cuda.serial
以上就是安装的全过程,后面会测试gpu串行能否进行,进行的速度如何。