Public Repository

Last pushed: a year ago
Short Description
Ubuntu Core 14.04 + CUDA 8.0 + cuDNN v5 + Theano.
Full Description

Background

I tried to run the following theano image with gtx 1080:

https://hub.docker.com/r/kaixhin/cuda-theano/

Unfortunately, faced two errors:

  1. ERROR (theano.sandbox.cuda): ERROR: Not using GPU. Initialisation of device gpu failed:
    initCnmem: cnmemInit call failed! Reason=CNMEM_STATUS_OUT_OF_MEMORY. numdev=1

  2. nvcc fatal : Value 'sm_61' is not defined for option 'gpu-architecture'

Error 1 can be solved by specify CNMEM=0.8 in the .theanorc
Error 2 can be solved by cuda-8.0 install based on:

http://stackoverflow.com/questions/38125310/nvcc-fatal-value-sm-61-is-not-defined-for-option-gpu-architecture-error-wi

Now, it works like a charm.

Usage

nvidia-docker run -it fananymi/theano:cuda-8.0

Docker Pull Command
Owner
fananymi

Comments (0)