无法初始化本机Deep Learning后端:null

时间:2017-06-23 11:43:48

标签: r machine-learning deep-learning nvidia h2o

所以,

我已经在我的ubuntu 16.4以及CUDA和CUDNN上安装了Nvidia驱动程序。我解决了所有问题,现在没关系。

它已经安装完毕,安装进展顺利。

我想在R中使用h2o.deepwater我的代码:

dlmodel <-  h2o.deepwater(x=predictors, y=response,
                         training_frame=train, 
                         validation_frame=valid, 

                         hidden=c(120,80), epochs=200, 
                         activation="Tanh",

                         seed=1234567,
                         stopping_metric='MSE',
                         classification_stop=0.001,
                         stopping_rounds = 5,
                         backend="tensorflow")

但我收到此错误java.lang.RuntimeException: Unable to initialize the native Deep Learning backend: null

任何人都可以帮我解决这个问题吗?我已经重新安装了R h2o和我的gpu驱动程序。

2 个答案:

答案 0 :(得分:2)

此常见原因是您尚未设置环境变量。请检查您的安装。

export CUDA_PATH=/usr/local/cuda export LD_LIBRARY_PATH=$CUDA_PATH/lib64:$LD_LIBRARY_PATH

此处进一步说明:https://github.com/h2oai/deepwater#pre-release-downloads

答案 1 :(得分:0)

请检查您是否正确设置了环境变量 - 但请注意:答案1和深水页(https://github.com/h2oai/deepwater#pre-release-downloads)中的片段在第二行中包含拼写错误。正确地说它应该是:

export CUDA_PATH=/usr/local/cuda
export LD_LIBRARY_PATH=$CUDA_PATH/lib64:$LD_LIBRARY_PATH