Как ускорить сверточную нейронную сеть GPU с помощью anano?

Question

Я использую theano для реализации нейронной сети свертки. Мое ОЗУ процессора - 32G, а ОЗУ GPU - 2G, но данные также очень большие - почти 5G данных обучения.

Когда программа запущена, компьютер, кажется, заморожен, и каждая операция выполняется очень медленно, даже не ответил. И режим процессора, по-видимому, не менее чем в 2 раза быстрее, чем режим GPU.

Есть ли способ ускорить сверточную нейронную сеть GPU?

Answer 1

Убедитесь в использовании Theano 0.7 с cudnn, эта скорость до свертка сильно:

http://deeplearning.net/software/theano/library/sandbox/cuda/dnn.html

Answer 2

Для того, чтобы использовать GPU accelleration первое, что вам нужно установить CUDA. На уровне конфигурации Theano (Theano флаги/TheanoRC) Есть несколько способов, вы можете ускорить вашу модель с GPU:

1. Specify использования GPU "DEVICE = ГП"
2. Включить выделение памяти Cuda (CnMem) "cnmem = 0,75"
3. Включить оптимизацию CUDNN "оптимизатор = cudnn"

Вы можете прочитать больше о Theano конфигурации here