Theano
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.