docker-torch-mega
Available from the Docker Hub as kaixhin/cuda-torch-mega
.
Ubuntu Core 14.04 + CUDA + cuDNN v5 + Torch7 (including iTorch). Includes the following additional packages:
- alewrap
- autograd
- classic
- dataset
- distlearn
- dpnn
- ffmpeg
- gvnn
- imagine-nn
- ipc
- loadcaffe
- logroll
- luaposix
- LuaSocket
- Moses
- nninit
- nnquery
- OptNet
- parallel
- rlenvs
- rnn
- signal
- tds
- torchnet
- torchx
- twrl
- xitari
Requirements
- NVIDIA Docker - see requirements for more details.
Usage
Use NVIDIA Docker: nvidia-docker run -it kaixhin/cuda-torch-mega
.
For more information on CUDA on Docker, see this readme.
To use Jupyter/iTorch open up the appropriate port. For example, use nvidia-docker run -it -p 8888:8888 kaixhin/cuda-torch-mega
. Then run jupyter notebook --ip="0.0.0.0" --no-browser
to open a notebook on localhost:8888
.