Tom Roth puzzler10
sparkydogX
/ occupy-memory.py
Last active
March 27, 2024 05:55
Pytorch trick : occupy all GPU memory in advance
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import os | |
| import torch | |
| from tqdm import tqdm | |
| import time | |
| # declare which gpu device to use | |
| cuda_device = '0' | |
| def check_mem(cuda_device): | |
| devices_info = os.popen('"/usr/bin/nvidia-smi" --query-gpu=memory.total,memory.used --format=csv,nounits,noheader').read().strip().split("\n") |
keitakurita
/ allennlp-basic.ipynb
Last active
July 10, 2020 00:25
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.