wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-05 13:21:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Autodetect the number of GPUs when starting Ray. (#1293)

Browse Source 
* autodetect

* Wed Dec  6 12:46:52 PST 2017

* Wed Dec  6 12:47:54 PST 2017

* Move GPU autodetection into services.py.

* Fix capitalization of Nvidia.

* Update documentation.
This commit is contained in:
Eric Liang
2017-12-09 15:30:16 -08:00
committed by Robert Nishihara
parent 6aae9a12fb
commit 7009538321
4 changed files with 23 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/worker.py
+3 -1
View File
@@ -1224,7 +1224,9 @@ def _init(address_info=None,
num_cpus (int): Number of cpus the user wishes all local schedulers to
be configured with.
num_gpus (int): Number of gpus the user wishes all local schedulers to
be configured with.
be configured with. If unspecified, Ray will attempt to autodetect
the number of GPUs available on the node (note that autodetection
currently only works for Nvidia GPUs).
resources: A dictionary mapping resource names to the quantity of that
resource available.
num_redis_shards: The number of Redis shards to start in addition to