我正在R中尝试keras。我尝试了以下代码作为Chollet的书。 已安装Cuda 9.0和Cudnn 7.4.2。
install.packages("keras")
library(keras)
install_keras(tensorflow="gpu")
imdb<-dataset_imdb(num_words = 10000)
但是它给出了错误消息(R会话异常终止,R遇到致命错误。会话已终止),我必须重新启动R 任何代码都会发生这种情况。
但是,如果我将tensorflow的增量更改为1.4,它会起作用。
install.packages("keras")
library(keras)
install_keras(tensorflow="1.4")
imdb<-dataset_imdb(num_words = 10000)
我应该怎么做才能解决这种终止现象? 为什么仅当option为gpu时才会发生这种终止。 预先感谢。