跑yolo训练时，用nvidia-smi查看我的显卡发现WDDM下面是ERR报错

Question

显存也只能跑一半，不知道是问题在哪
1050ti的显卡
请教大家，初学，不太确定是哪里的错误
是我的cudann版本没装对吗？

Answer 1

让阿豪来帮你解答，本回答参考chatgpt3.5编写提供，如果还有疑问可以评论或留言

问题描述：

• 使用1050ti的显卡跑程序，发现显存只能跑一半。
• 不确定错误出在哪里，猜测可能是cudann版本没装对。 回答：
• 首先，显存只能跑一半可能是因为程序中存在某些内存泄漏或者未释放的显存。
• 其次，显存的问题并不一定与cudnn版本相关，可以先排除这个因素。
• 如果确定是代码问题导致的显存问题，可以考虑进行一些优化，比如减少显存占用等。
• 对于显存占用问题，可以使用Tensorflow和PyTorch中的with tf.device('gpu:0')和torch.cuda.empty_cache()等方法来控制显存的使用和释放。
• 下面给出一个简单的代码示例，使用PyTorch来控制显存使用和释放：

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()