{ "info": { "author": "Zhang Tianyu", "author_email": "tyz.xyz@qq.com", "bugtrack_url": null, "classifiers": [ "Development Status :: 3 - Alpha", "Intended Audience :: Developers", "License :: OSI Approved :: MIT License", "Programming Language :: Python :: 3" ], "description": "# gpuinfo\n\nI implement some functions that can help users to obtain nvidia gpu information.\n\nTo use gpuinfo, you need to be able to run 'ps' and 'nvidia-smi' in your terminal. \n\n# Install with pip\n```\npip install gpuinfo\n``` \nI only tested on linux system with python3.\nhttps://pypi.org/project/gpuinfo/\n\n# Usage\n\n```python\nfrom gpuinfo import GPUInfo\n```\n\nGPUInfo has the following functions:\n\n get_users(gpu_id)\n return a dict. show every user and memory on a certain gpu \n\n check_empty()\n check_empty()\n return a list containing all GPU ids that no process is using currently.\n\n get_info()\n pid_list,percent,memory,gpu_used=get_info()\n return a dict and three lists. pid_list has pids as keys and gpu ids as values, showing which gpu the process is using\n\n get_user(pid)\n get_user(pid)\n Input a pid number , return its creator by linux command ps\n\n gpu_usage()\n gpu_usage()\n return two lists. The first list contains usage percent of every GPU. The second list contains the memory used of every GPU. The information is obtained by command 'nvidia-smi'\n\n# Example\n\n```python\nfrom gpuinfo import GPUInfo\n\navailable_device=GPUInfo.check_empty()\n#available_device\u5c31\u662f\u4e00\u4e2a\u542b\u6709\u6240\u6709\u6ca1\u6709\u4efb\u52a1\u7684gpu\u7f16\u53f7\u7684\u5217\u8868\npercent,memory=GPUInfo.gpu_usage()\n#\u83b7\u5f97\u6240\u6709gpu\u7684\u4f7f\u7528\u767e\u5206\u6bd4\u548c\u663e\u5b58\u5360\u7528\u91cf\nmin_percent=percent.index(min([percent[i] for i in available_device]))\n#\u672a\u88ab\u4f7f\u7528\u7684gpu\u91ccpercent\u6700\u5c0f\u7684\nmin_memory=memory.index(min([memory[i] for i in available_device]))\n#\u672a\u88ab\u4f7f\u7528\u7684gpu\u91cc\u663e\u5b58\u5360\u7528\u91cf\u6700\u5c11\u7684\n\n#\u5982\u679c\u4f60\u4f7f\u7528pytorch\ntorch.cuda.set_device(min_percent) \u6216\u8005(min_memory)\n```\n\n\n", "description_content_type": "text/plain", "docs_url": null, "download_url": "", "downloads": { "last_day": -1, "last_month": -1, "last_week": -1 }, "home_page": "", "keywords": "gpu information", "license": "", "maintainer": "", "maintainer_email": "", "name": "gpuinfo", "package_url": "https://pypi.org/project/gpuinfo/", "platform": "", "project_url": "https://pypi.org/project/gpuinfo/", "project_urls": { "Source": "https://github.com/FlyHighest/gpuinfo/" }, "release_url": "https://pypi.org/project/gpuinfo/1.0.0a7/", "requires_dist": null, "requires_python": "", "summary": "A quick access to nvidia gpu information", "version": "1.0.0a7", "yanked": false, "yanked_reason": null }, "last_serial": 8347004, "releases": { "1.0.0a1": [ { "comment_text": "", "digests": { "md5": "75540f5f7b7a55d36cf345f4d69e3f3f", "sha256": "6a6fa4110e2168ae7eff00d874cd6dd6b54aaef781dbbaafdd42331dfe4c903b" }, "downloads": -1, "filename": "gpuinfo-1.0.0a1.tar.gz", "has_sig": false, "md5_digest": "75540f5f7b7a55d36cf345f4d69e3f3f", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 4555, "upload_time": "2018-04-18T08:45:15", "upload_time_iso_8601": "2018-04-18T08:45:15.338131Z", "url": "https://files.pythonhosted.org/packages/e5/53/3c1dcdb5a828695b825d98312ca11057bb33d929babe414074b58c797b8a/gpuinfo-1.0.0a1.tar.gz", "yanked": false, "yanked_reason": null } ], "1.0.0a2": [ { "comment_text": "", "digests": { "md5": "993ed5c3f0db78f79031eff4f7070ec0", "sha256": "93f84a13f4d91bb392b32a24e41db9c5bdf1c072d5ae1c694f7e74277e5532c3" }, "downloads": -1, "filename": "gpuinfo-1.0.0a2-py3-none-any.whl", "has_sig": false, "md5_digest": "993ed5c3f0db78f79031eff4f7070ec0", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 3081, "upload_time": "2018-06-15T06:53:22", "upload_time_iso_8601": "2018-06-15T06:53:22.069441Z", "url": "https://files.pythonhosted.org/packages/10/16/a541cf2da21d5f48e46d3ad42b678df0022d300500958b7bd2b53747c93a/gpuinfo-1.0.0a2-py3-none-any.whl", "yanked": false, "yanked_reason": null }, { "comment_text": "", "digests": { "md5": "bca4f22dd4a8e3cfe0d530a9ec730b8f", "sha256": "1caac61c52bcebab6c700757cea95953ecfed09afa544343c758727cf1f5657b" }, "downloads": -1, "filename": "gpuinfo-1.0.0a2.tar.gz", "has_sig": false, "md5_digest": "bca4f22dd4a8e3cfe0d530a9ec730b8f", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5056, "upload_time": "2018-06-15T06:53:23", "upload_time_iso_8601": "2018-06-15T06:53:23.683246Z", "url": "https://files.pythonhosted.org/packages/bd/58/84b0c0a89b505b7f2120c5e052bab9c844733cecd1e9ddaa5c53cec830b0/gpuinfo-1.0.0a2.tar.gz", "yanked": false, "yanked_reason": null } ], "1.0.0a3": [ { "comment_text": "", "digests": { "md5": "6624894737e2d78356abd6cb75d39c22", "sha256": "146918e5027999b74e0a052313286dcd309d9b8b4884379a0e9063af52b51191" }, "downloads": -1, "filename": "gpuinfo-1.0.0a3-py3-none-any.whl", "has_sig": false, "md5_digest": "6624894737e2d78356abd6cb75d39c22", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 3242, "upload_time": "2018-06-22T08:06:56", "upload_time_iso_8601": "2018-06-22T08:06:56.392492Z", "url": "https://files.pythonhosted.org/packages/a4/c9/3b4fa155874f0f3b1701d5ddba1d89ac5749160ca946aaf35f8cf60f7d6f/gpuinfo-1.0.0a3-py3-none-any.whl", "yanked": false, "yanked_reason": null }, { "comment_text": "", "digests": { "md5": "979d98ee898cf6a2192ac71b5fc167ef", "sha256": "d495f8fd8bd543188dfbae125fe9b84f7d26a25122694bf0289aac1108fd71f7" }, "downloads": -1, "filename": "gpuinfo-1.0.0a3.tar.gz", "has_sig": false, "md5_digest": "979d98ee898cf6a2192ac71b5fc167ef", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5206, "upload_time": "2018-06-22T08:06:57", "upload_time_iso_8601": "2018-06-22T08:06:57.630866Z", "url": "https://files.pythonhosted.org/packages/95/e2/aacf3d0598fb05774748302d2defa2a95bcbb7075d9f0129e81508015583/gpuinfo-1.0.0a3.tar.gz", "yanked": false, "yanked_reason": null } ], "1.0.0a4": [ { "comment_text": "", "digests": { "md5": "3d68452bb3cb0874ce01003be7a037b7", "sha256": "a190a26007670a30cf4afffcb9681510efd5844f43cfc84ac37178f2c34eb3a0" }, "downloads": -1, "filename": "gpuinfo-1.0.0a4-py3-none-any.whl", "has_sig": false, "md5_digest": "3d68452bb3cb0874ce01003be7a037b7", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 3251, "upload_time": "2018-06-22T08:17:07", "upload_time_iso_8601": "2018-06-22T08:17:07.276671Z", "url": "https://files.pythonhosted.org/packages/cd/68/5a53902e614517813b5c967f7893478d9840c5eeda16689c7d3174a86120/gpuinfo-1.0.0a4-py3-none-any.whl", "yanked": false, "yanked_reason": null }, { "comment_text": "", "digests": { "md5": "a430a8d7265943ca38ba259b0bfef1e1", "sha256": "dfdbc32a129186c01ba821ff17026d9de30b24349e8c1bfa46712ffd7337fc96" }, "downloads": -1, "filename": "gpuinfo-1.0.0a4.tar.gz", "has_sig": false, "md5_digest": "a430a8d7265943ca38ba259b0bfef1e1", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5217, "upload_time": "2018-06-22T08:17:08", "upload_time_iso_8601": "2018-06-22T08:17:08.487132Z", "url": "https://files.pythonhosted.org/packages/24/7c/0c7ae87be97697ba837d0ab8a458e4c08047cdfc8e6f4b3412a921dddd95/gpuinfo-1.0.0a4.tar.gz", "yanked": false, "yanked_reason": null } ], "1.0.0a5": [ { "comment_text": "", "digests": { "md5": "f275df909691cc306d8640f837c2d796", "sha256": "5f598ef2c111963b2ccc91966d15a1349c000a65c5f5378aea18ea8cdd66991d" }, "downloads": -1, "filename": "gpuinfo-1.0.0a5-py3.7.egg", "has_sig": false, "md5_digest": "f275df909691cc306d8640f837c2d796", "packagetype": "bdist_egg", "python_version": "3.7", "requires_python": null, "size": 5025, "upload_time": "2019-05-25T06:52:40", "upload_time_iso_8601": "2019-05-25T06:52:40.300472Z", "url": "https://files.pythonhosted.org/packages/ab/32/1e51fcb9535a10293a1b88bab876dd1caa6a958c750ad3c429544b716042/gpuinfo-1.0.0a5-py3.7.egg", "yanked": false, "yanked_reason": null }, { "comment_text": "", "digests": { "md5": "faf0ff11689ae819802bbd9b537aad86", "sha256": "de7f48c34509834f733a59b606083d9a6119df2988f8b4cc9b0331d11c997a43" }, "downloads": -1, "filename": "gpuinfo-1.0.0a5.tar.gz", "has_sig": false, "md5_digest": "faf0ff11689ae819802bbd9b537aad86", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5257, "upload_time": "2019-05-25T06:52:41", "upload_time_iso_8601": "2019-05-25T06:52:41.924933Z", "url": "https://files.pythonhosted.org/packages/c4/71/fc4f6bf6e4d0e1ed8d16e84f33cbebe3464bc1377edcce29f6cb6fff1d6e/gpuinfo-1.0.0a5.tar.gz", "yanked": false, "yanked_reason": null } ], "1.0.0a6": [ { "comment_text": "", "digests": { "md5": "76cbc060b30bcf075c063ef6d1fdf94a", "sha256": "592b88504c78c2553d39725308ffd092d1045323af9f75c484db45fe58f8448a" }, "downloads": -1, "filename": "gpuinfo-1.0.0a6-py3-none-any.whl", "has_sig": false, "md5_digest": "76cbc060b30bcf075c063ef6d1fdf94a", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 4071, "upload_time": "2019-10-29T03:11:28", "upload_time_iso_8601": "2019-10-29T03:11:28.320723Z", "url": "https://files.pythonhosted.org/packages/45/c6/70fc3d7c04b177575dba2f74a1490484860bceaafeb4d3fb7f313e4208e6/gpuinfo-1.0.0a6-py3-none-any.whl", "yanked": false, "yanked_reason": null }, { "comment_text": "", "digests": { "md5": "e7f894466fb562a10971c8324a3a5670", "sha256": "35536cb0b9c4f62d9bbf7e0da7f040e418421337763808fb5f350bbe902f4b83" }, "downloads": -1, "filename": "gpuinfo-1.0.0a6.tar.gz", "has_sig": false, "md5_digest": "e7f894466fb562a10971c8324a3a5670", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5260, "upload_time": "2019-10-29T03:11:29", "upload_time_iso_8601": "2019-10-29T03:11:29.931355Z", "url": "https://files.pythonhosted.org/packages/71/ce/d7305c2479a4d262a14fee9facaadfe21645790390a77210665ab37a94dd/gpuinfo-1.0.0a6.tar.gz", "yanked": false, "yanked_reason": null } ], "1.0.0a7": [ { "comment_text": "", "digests": { "md5": "5e1b5a4d6bc2ff8d9cfdab99605d3e1f", "sha256": "a0dc7191542c95599e807ae103e80d63138fa82a472d0b7d22e757208586e6ab" }, "downloads": -1, "filename": "gpuinfo-1.0.0a7-py3-none-any.whl", "has_sig": false, "md5_digest": "5e1b5a4d6bc2ff8d9cfdab99605d3e1f", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 4086, "upload_time": "2020-10-06T02:32:17", "upload_time_iso_8601": "2020-10-06T02:32:17.329479Z", "url": "https://files.pythonhosted.org/packages/64/c8/bb3e935843f583fbf9970e125b888a2c7c0ec6ce6eada8d3f69d88be0294/gpuinfo-1.0.0a7-py3-none-any.whl", "yanked": false, "yanked_reason": null }, { "comment_text": "", "digests": { "md5": "f10418f11882f80b75dfb10bcfa0b051", "sha256": "56377301dbbd09c22b1bc19d5bb5d291b835676ed8e95f9004c195a091c7a077" }, "downloads": -1, "filename": "gpuinfo-1.0.0a7.tar.gz", "has_sig": false, "md5_digest": "f10418f11882f80b75dfb10bcfa0b051", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5265, "upload_time": "2020-10-06T02:32:19", "upload_time_iso_8601": "2020-10-06T02:32:19.093235Z", "url": "https://files.pythonhosted.org/packages/03/94/0d99b8b788cd770cbd85f9324d874208c350f6c6d91596374a612ae558d3/gpuinfo-1.0.0a7.tar.gz", "yanked": false, "yanked_reason": null } ] }, "urls": [ { "comment_text": "", "digests": { "md5": "5e1b5a4d6bc2ff8d9cfdab99605d3e1f", "sha256": "a0dc7191542c95599e807ae103e80d63138fa82a472d0b7d22e757208586e6ab" }, "downloads": -1, "filename": "gpuinfo-1.0.0a7-py3-none-any.whl", "has_sig": false, "md5_digest": "5e1b5a4d6bc2ff8d9cfdab99605d3e1f", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 4086, "upload_time": "2020-10-06T02:32:17", "upload_time_iso_8601": "2020-10-06T02:32:17.329479Z", "url": "https://files.pythonhosted.org/packages/64/c8/bb3e935843f583fbf9970e125b888a2c7c0ec6ce6eada8d3f69d88be0294/gpuinfo-1.0.0a7-py3-none-any.whl", "yanked": false, "yanked_reason": null }, { "comment_text": "", "digests": { "md5": "f10418f11882f80b75dfb10bcfa0b051", "sha256": "56377301dbbd09c22b1bc19d5bb5d291b835676ed8e95f9004c195a091c7a077" }, "downloads": -1, "filename": "gpuinfo-1.0.0a7.tar.gz", "has_sig": false, "md5_digest": "f10418f11882f80b75dfb10bcfa0b051", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 5265, "upload_time": "2020-10-06T02:32:19", "upload_time_iso_8601": "2020-10-06T02:32:19.093235Z", "url": "https://files.pythonhosted.org/packages/03/94/0d99b8b788cd770cbd85f9324d874208c350f6c6d91596374a612ae558d3/gpuinfo-1.0.0a7.tar.gz", "yanked": false, "yanked_reason": null } ], "vulnerabilities": [] }