add option for selection of cuda devices

uw-ipd / RoseTTAFold2

MIT License

170 stars 39 forks source link

add option for selection of cuda devices #16

Open Cynicjon opened 1 year ago

Cynicjon commented 1 year ago

Adds a command line option, --cuda, that takes an integer corresponding to the cuda device to run on, and sets the CUDA_VISIBLE_DEVICES environment variable accordingly.

re: https://github.com/uw-ipd/RoseTTAFold2/issues/15