Mathematica中文论坛-非官方

 找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
热搜: 活动 交友 discuz
查看: 11344|回复: 1
打印 上一主题 下一主题

ubuntu install Deep Tensor Neural Networks

[复制链接]

529

主题

598

帖子

3014

积分

论坛元老

Rank: 8Rank: 8

积分
3014
跳转到指定楼层
楼主
发表于 2017-12-1 14:03:16 | 只看该作者 回帖奖励 |倒序浏览 |阅读模式
本帖最后由 meatball1982 于 2017-12-1 14:42 编辑

看NC的一个文儿。觉得蛮好的(老板觉得蛮好的)
K.T. Schütt. F. Arbabzadah. S. Chmiela, K.-R. Müller, A. Tkatchenko.
Quantum-chemical insights from deep tensor neural networks.

Nature Communications 8. 13890 (2017)
doi: 10.1038/ncomms13890


重复他的工作。
具体干什么的,还得一点点的看。
把环境build起来先。因为作者把它的工作都给你了。剩下的,就是重新来一下。
具体的,在

https://github.com/atomistic-machine-learning/dtnn



要求:
Requirements:
  • python 3.4
  • ASE
  • numpy
  • tensorflow (>=1.0)
你M,除了numpy(实际上我的是python2.7下的,在python3.4下的,我也是不满足),其它的都不满足。

整!!!!
花了我整整三天的时间,从一点点的安装,到重新安装,再安装,安装,安装。。。。。。
我操。


因为自己的机器有cuda7.5 和cuda-9.0
要求的是cuda8.0 ,
sudo ln -s /usr/local/cuda-9.0/lib64/libcublas.0.9.0 /usr/local/cuda-9.0/lib64/libcublas.0.8.0

没少用(特别是在import tensorflow as tf)的时候,它告诉你,少libcublas.0.8.0,你就在0.9.0那里放一个link。
我的机器是python2.7

安装pip3先。
sudo atp-get install pip3

告诉你要安装一个我也记不起来的
好像是 python3-pip之类的。这个得sudo

然后是
pip3 install ase
这个花的时间挺长,但会把numpy也给安装了。
GPU:
pip3 install tensorflow-gpu
注意,要是用GPU跑,就用这个。


CPU
pip3 install tensorflow-cpu

会安装一堆东西,和 tensorflow官网让你安装的差不多(这个先题是你的cuda,cudnn都安装完了)
然后
python3
进来后
import tensorflow as tf
tf.__version__
注意,是两个_
我的tf是1.4
我的python3是3.5

在跑例子的时候,work_train.py就是rain_dtnn_gdb9.py

CUDA_VISIBLE_DEVICES="0,1" python3 work_train.py ../dtnn/datasets/ ../out_res/

要在import dtnn前加
import sys
sys.path.append('/the path where u put you dtnn/dtnn-master/')
import dtnn

CUDA_VISIBLE_DEVICES是指定你用哪个GPU,实验室华丽的K80,我和DL一人一半。

而我的w530,16G内存,2G显存,是跑不起来的。

当看到在训练的时候,TM差点哭出来。
记录下来,有问题问meatball1982@163.com






dtnn-master (1).zip

19.46 KB, 下载次数: 0

分享到:  QQ好友和群QQ好友和群 QQ空间QQ空间 腾讯微博腾讯微博 腾讯朋友腾讯朋友
收藏收藏
回复

使用道具 举报

529

主题

598

帖子

3014

积分

论坛元老

Rank: 8Rank: 8

积分
3014
沙发
 楼主| 发表于 2019-7-13 14:02:37 | 只看该作者
如果不行,可以全程使用conda 进行安装 。
conda install cudatoolkit
conda install cudnn
(安装特定版本的toolkit)conda install cudatoolkit==10.0.0
conda install tensorflow-gpu
conda install -c conda-forge ase

conda真是好啊。

回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|Archiver|手机版|小黑屋|Mathematica中文论坛-非官方 ( 辽ICP备16001491号-1

GMT+8, 2024-4-20 18:18 , Processed in 0.125641 second(s), 25 queries .

Powered by Discuz! X3.2

© 2001-2013 Comsenz Inc.

快速回复 返回顶部 返回列表