Компьютерная лингвистика

Новостная лента www.solarix.ru

Previous Entry Share Next Entry
Theano 1.0.1 с cuda 9.0 поломан?
kelijah
Что-то с новым релизом Theano 1.0.1 нечисто. По крайней мере при работе с Cuda 9.0 под убунтой и вторым питоном.

Уже на второй модели вижу, что сетка отказывается обучаться - тупо loss на валидации стоит на одном значении с точностью до 4го знака несколько эпох подряд и баста, привет early stopping.

Версии:
inkoziev@Ubuntu-1604-xenial-64-minimal:~/polygon/AttrExtractor_srv/py$ python
Python 2.7.12 (default, Dec  4 2017, 14:50:18)
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import theano
Using cuDNN version 7101 on context None
Mapped name None to device cuda0: GeForce GTX 1080 (0000:01:00.0)
>>> print theano.__version__
1.0.1
>>>


При этом на этой же машине последний Tensorflow дает замечательные результаты для той же самой модели при запуске с KERAS_BACKEND=tensorflow. Хотя битва с tf у меня была длительная из-за того, что Cuda 9.1 не поддерживается, так что пришлось выбирать между компиляцией TF из исходников и даунгрейдом до cuda 9.0, я выбрал cuda 9.0 и не ошибся. А теперь точит червь сомнения насчет Theano.

?

Log in

No account? Create an account