2022年1月19日
卡GPU显存小程序
摘要： 使用之前只需要简单的设置两个变量即可： Memory_Allocated_GB = 5 # GiB，需要占用的显存大小，单位GiB GPU_ID = 2 # 需要卡显存的GPU ID import torch from torch import cuda import time import psu 阅读全文
posted @ 2022-01-19 00:49 玖玖牛 阅读(259) 评论(0) 推荐(0)