Theano

Z MetaCentrum
Skočit na navigaci Skočit na vyhledávání


Description

Theano is a Python library that allows you to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays efficiently.

Theano features:

  • tight integration with NumPy
  • transparent use of a GPU
  • efficient symbolic differentiation
  • speed and stability optimizations

Availability

  • Theano version 1.0.4: module python36-modules-gcc
  • Theano version 0.7: module theano-0.7
  • Theano version 0.6: module theano-0.6

Usage

Load the module by running:

module add python36-modules-gcc

Now you can in your python scripts call

import theano

Use with CUDA

In general you can use Theano module on GPUs. See the thorough documentation or use this shortened guideline.

  • Plan your job to some GPU machine: qsub -q gpu -l select=1:ncpus=1:ngpus=X:gpu_cap=cuda61 <job batch file>
  • Enable CUDA computing (Note: you have to be registered in NVIDIA developer program to be able to use CuDNN):
 export THEANO_FLAGS='device=cuda,floatX=float32'
 module add cudnn-7.6.4-cuda10.1
  • Run Theano: python -c "import theano; import theano.gpuarray; print(theano.__version__)"

License

BSD license

Documentation

The Theano documentation is available at official webpage.

Homepage

URL: http://deeplearning.net/software/theano/