Skip to content

require cudnn takes 10 minutes on a Volta with 1 GPU (Cuda 9, cudnn 7.1) #394

Description

@ajhool

The following command takes 10 minutes:

    #mycode.lua
    print('params.backend == cudnn, require cudnn')
    require 'cudnn'
    print('cudnn required')

require nn takes a split second.

And am building it using docker:

FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04 as base
...
RUN git clone https://github.com/soumith/cudnn.torch.git -b R7 && cd cudnn.torch && \
    luarocks make cudnn-scm-1.rockspec

#385 also talks about a slow loading time due to gpu initialization. Does setting the # of visible GPUs to zero affect runtime?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions