DeepMD安装及Unbuntu下Nvidia显卡升级（防跳坑）

Krow

于 2020-08-07 17:03:53 发布

阅读量2.8k

点赞数

分类专栏： DeepMD Linux学习笔记文章标签： ubuntu 驱动程序 tensorflow 神经网络

本文链接：https://blog.youkuaiyun.com/weixin_43120346/article/details/107864695

版权

本文档详细记录了在Ubuntu系统下安装DeepMD机器学习MD工具包的过程，以及遇到的CUDA版本与Nvidia驱动不匹配的问题。通过升级Nvidia驱动至450.57版本，成功解决Tensorflow运行错误。此外，提到了在结束神经网络训练后，导入Lammps进行MD可能遇到的核心转储错误，并给出了解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近由于需要重新使用DeepMD，实验室服务器又崩了，/opt /home阵列全完蛋了，只能在自己的机器上进行重新安装。
DeepMD是由Wang等人开发的机器学习MD工具包，其github地址如下：

https://github.com/deepmodeling/deepmd-kit/

我使用了DeepMD官方推荐的conda方法安装，不得不说conda真是神器，安装早期版本的时候跳了一万次坑。

此处选用GPU版本，推荐conda虚拟环境，所有需要GPU的包都强烈建议使用虚拟环境运行，一面出现差错。此处我使用的anaconda版本为4.6.11，新版本可能出现错误。

conda create -n dpmd python=3.6 #建立虚拟环境
source activate dpmd #激活虚拟环境 对于新版本conda 使用 conda activate dpmd
conda install deepmd-kit=*=*gpu lammps-dp=*=

最低0.47元/天解锁文章