python自动选取空闲gpu

自动选取可用gpu

pip install nvidia-ml-py==4.304.04

pip安装pynvml pip install nvidia-ml-py3

如果是Python2，则pip nvidia-ml-py2

import os
import pynvml
pynvml.nvmlInit()def usegpu(need_gpu_count=1):nouse=[]for index in range(pynvml.nvmlDeviceGetCount()):# 这里的0是GPU idhandle = pynvml.nvmlDeviceGetHandleByIndex(index)meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)used= meminfo.used/meminfo.totalif used<0.5:nouse.append(index)if len(nouse)>=need_gpu_count:os.environ["CUDA_VISIBLE_DEVICES"] = ','.join(map(str, nouse[:need_gpu_count]))return nouse[:need_gpu_count]elif len(nouse)>0:os.environ["CUDA_VISIBLE_DEVICES"] = ','.join(map(str, nouse))return len(nouse)else:return 0if __name__ == '__main__':gpus=usegpu(need_gpu_count=2)if gpus:print("use gpu ok")else:print("no gpu is valid")

python自动选取空闲gpu相关推荐

pytorch 指定卡1_如何为TensorFlow和PyTorch自动选择空闲GPU，解决抢卡争端
原标题:如何为TensorFlow和PyTorch自动选择空闲GPU,解决抢卡争端雷锋网按:本文作者天清,原文载于其知乎专栏世界那么大我想写代码,雷锋网获其授权发布. 项目地址:QuantumLiu ...
python自动答题免费_直播答题？Python助你自动搜题之新手篇！
承接前文关于Python自动搜题的介绍https://zhuanlan.zhihu.com/p/32828411,此篇面向新手小白进行解析,助你成功运行自动搜题的python程序. A部分是介绍思路, ...
用Python自动生成数据日报！
今天聊聊怎么用Python自动生成数据日报! 其实我觉得蛮简单,核心就是你组装好日报的内容模板,然后将变化的量交给python去填充,需要用到的基本就是python处理excel.word和ppt等相 ...
Python+PyCharm+PyTorch+Cuda/GPU 安装步骤
Python+PyCharm+PyTorch+Cuda/GPU 安装步骤一. 无nvidia显卡,直接用CPU跑程序时二.有Nvidia显卡,想用GPU跑程序时 2.1 先查版本! 2.2 下载安 ...
Python自动生成数据日报
对于一个企业来说,高层看意义,中层看结论,基层看落地,数据日报.周报.月报可以监控销售个人在实际执行过程中的销售动态,而数据季度报.年报可以反映一个销售策略是否与实际的业务场景切合. 可见数据日报在我 ...
python自动答题软件_直播答题？Python助你自动搜题之新手篇！
承接前文关于Python自动搜题的介绍 https:// zhuanlan.zhihu.com/p/32 828411 ,此篇面向新手小白进行解析,助你成功运行自动搜题的python程序. A部分是介 ...
Python 之vim编写python自动补全
Pydiction :vim - python自动补全插件插件的安装如下: 1.下载插件包 https://github.com/vim-scripts/Pydiction 可以直接下载,也可git ...
python资料百度网盘-python自动保存百度盘资源到百度盘中的实例代码
本实例的实现逻辑是,应用selenium UI自动化登录百度盘,读取存储百度分享地址和提取码的txt文档,打开百度盘分享地址,填入提取码,然后保存到指定的目录中全部代码如下: # -*-coding ...
python画轨迹曲线-python 自动轨迹绘制的实例代码
用到的思维: 自动化思维,数据和功能分开处理,用数据驱动程序自动运行接口化设计,数据与程序的对接方式要清晰明了二维数据应用,应用维度组织数据,二维数据最常用代码 # AutoTrace.py i ...

python自动选取空闲gpu

python自动选取空闲gpu相关推荐

最新文章

热门文章