r - 无法初始化 native 深度学习后端: null

Question

所以，

我已经在我的 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:无法初始化 native 深度学习后端:null

谁能帮我解决这个问题吗？我已经重新安装了 R h2o 和我的 GPU 驱动程序。

Answer 1

造成这种情况的一个常见原因是您尚未设置环境变量。请检查您的安装。

导出 CUDA_PATH=/usr/local/cuda 导出 LD_LIBRARY_PATH=$CUDA_PATH/lib64:$LD_LIBRARY_PATH

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