我正在使用Keras和Theano作为后端,我有顺序神经网络模型.
我想知道以下是否有区别?
model.add(Convolution2D(32, 3, 3, activation='relu'))
Run Code Online (Sandbox Code Playgroud)
和
model.add(Convolution2D(32, 3, 3))
model.add(Activation('relu'))
Run Code Online (Sandbox Code Playgroud) 我试图使用 CUDA API 以 GPU 加速运行暗网。所以我按照这里的说明,在 Makefile 中更改 GPU=1 并开始制作。当我尝试运行测试时,由于 CUDA 错误而失败。
./darknet yolo test cfg/yolo.cfg yolo.weights data/dog.jpg CUDA 错误:未知错误暗网:./src/cuda.c:21:check_error:断言“0”失败。
我在 Titan X 上使用 Ubuntu 14.04、CUDA 7.5 和我的 NVIDIA-SMI 352.93 和驱动程序版本:352.93 我很确定我的 CUDA 工作正常并且驱动程序的版本是最新的,因为我用它来加速 Caffe . 我的猜测是暗网找不到 CUDA 目录。
谁能帮我解决这个问题?