search

MadhumitaSushil / SDAE

Stacked Denoising AutoEncoder
GNU General Public License v3.0
74 stars 20 forks source link

Theano #4

Open Missshulan opened 4 years ago

Missshulan commented 4 years ago

I ran your code, and the following error was prompted，“ValueError: You are tring to use the old GPU back-end. It was removed from Theano. Use device=cuda* now. See https://github.com/Theano/Theano/wiki/Converting-to-the-new-gpu-back-end%28gpuarray%29 for more information.”. How can I modify it？

sfdk74 commented 4 years ago

I have this issue too and solved this by changing next line in sdae.py: os.environ['THEANO_FLAGS'] = "device=gpu1,floatX=float32" to: os.environ['THEANO_FLAGS'] = "device=cuda*,floatX=float32"