GPU usage

[kudpolser]

Dear developers,

DeepRescore does not use the GPU on our server. How do I get him to see our graphic cards?
I assume I need to pass the parameters to Docker --runtime=nvidia --NVIDIA_VISIBLE_DEVICES=all, but due to the fact that I am using nextflow, I don't understand how to do this.

Thanks in advance.

Polina

[wenbostar]

Have you installed https://github.com/NVIDIA/nvidia-docker?

Bo

[nukaemon]

Hello @kudpolser

I encountered with the same problem when running on AWS server.
Add '--gpus all' to nextflow.config like as

docker {

        enabled    = true
        remove     = true
        runOptions = '-u $(id -u):$(id -g) --gpus all'
        temp       = "/tmp/"
}

then you should pass train_autoRT process.