显存也只能跑一半,不知道是问题在哪
1050ti的显卡
请教大家,初学,不太确定是哪里的错误
是我的cudann版本没装对吗?
让阿豪来帮你解答,本回答参考chatgpt3.5编写提供,如果还有疑问可以评论或留言问题描述:
import torch
# 指定使用的gpu设备
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
# 使用with语句控制显存使用
with torch.cuda.device(device):
x = torch.ones((1024, 1024), device=device)
y = torch.ones((1024, 1024), device=device)
z = x + y
# 释放显存
torch.cuda.empty_cache()