fastai / courses

fast.ai Courses

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

install_gpu.sh: ValueError «You are trying to use the old GPU back-end» when importing

nullbyte91 opened this issue · comments

Host Information:
Ubuntu 16.04 LTS
Graphics Quadro P2000

conda 4.5.12
theano 1.0.3

Using the install_gpu.sh to set up the fast.ai env and When I ran the deeplearning1/nbs/lesson1.ipynb notebook getting the below error,
ValueError: You are tring to use the old GPU back-end

I could able to resolve this issue by exporting THEANO_FLAGS.
$ export THEANO_FLAGS=mode=FAST_RUN,device=cuda,floatX=float32

If this's valid one, We have to incorporate this changes to install_gpu.sh.