您的位置:首页 > 财经 > 产业 > WSL中使用AMBER GPU串行版

WSL中使用AMBER GPU串行版

2024/12/23 9:11:34 来源:https://blog.csdn.net/handsome213/article/details/142287333  浏览:    关键词:WSL中使用AMBER GPU串行版

        前提是已经安装过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 $AMBERHOMEexport CUDA_VISIBLE_DEVICES=0 (choose the GPU id you wish to test)make test.cuda.serial

        以上就是安装的全过程,后面会测试gpu串行能否进行,进行的速度如何。

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com