Skip to content

A quick tutorial on how to unpickle GPU saved variables, on a CPU machine.

Notifications You must be signed in to change notification settings

Rostifar/TheanoUnpickler

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

TheanoUnpickler

A quick tutorial on how to unpickle GPU saved variables, on a CPU machine.

The key to unpickling these variables is setting theano.config.experimental.unpickle_gpu_on_cpu = True

  • This allows the user to unpickle these variables without theano throwing a CUDA exception.
  • Because theano variables are often stored as CudaNdArrays you often need a GPU to unpickle the parameters of previously trained models

I have included a small python script which i used to convert CudaNdArrays to NumpyArrays when unpickling

If you have any questions contact me @: +rcbridendev@gmail.com

About

A quick tutorial on how to unpickle GPU saved variables, on a CPU machine.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages