소위 말해 python 안에서 nvidia-smi 를 수행하는 방법이다. pip install nvidia-ml-py3 위 패키지를 설치하고 아래와 같이 패키지를 불러와 사용한다. import nvidia_smi nvidia_smi.nvmlInit() deviceCount = nvidia_smi.nvmlDeviceGetCount() freem = [] for i in range(deviceCount): handle = nvidia_smi.nvmlDeviceGetHandleByIndex(i) info = nvidia_smi.nvmlDeviceGetMemoryInfo(handle) freem.append(100*info.free/info.total) nvidia_smi.nvmlShutdown() 이제 fre..

연구/정보 2022. 3. 22. 01:23

결론부터. s = [12, 13, 11, 14, 15, 13] sorted(range(len(s)), key=lambda k: s[k]) >>> [2, 0, 1, 5, 3, 4] 데이터를 다루다보면 단순히 리스트를 정렬하는 게 아닌 리스트가 가진 값들을 가리키는 인덱스를 순서대로 정렬해야 할 때가 많다. 이럴 때 위와 같이 코드 한 줄로 해결한다. 거꾸로 정렬 하고 싶다면 아래와 같이 sorted 함수의 reverse=True로 설정한다. s = [12, 13, 11, 14, 15, 13] sorted(range(len(s)), key=lambda k: s[k], reverse=True) >>> [4, 3, 5, 1, 0, 2] 출처 https://stackoverflow.com/questions/785..

연구/정보 2020. 11. 4. 14:46