所以,
我已经在我的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驱动程序。
答案 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