how to use muti gpus or choice the gpu that i want to use? - Githubissues

dptech-corp / Uni-Dock

Uni-Dock: a GPU-accelerated molecular docking program

189 stars 39 forks source link

how to use muti gpus or choice the gpu that i want to use? #15

Closed mapengsen closed 1 year ago

mapengsen commented 1 year ago

how to use muti gpus or choice the gpu that i want to use?

caic99 commented 1 year ago

Hi @mapengsen , Please set environment variable CUDA_VISIBLE_DEVICES before running unidock, e.g. CUDA_VISIBLE_DEVICES=2 unidock <params>. Please note that one unidock process does not support more than one GPU.