指定使用哪块GPU运行程序

最新推荐文章于 2024-09-01 16:56:02 发布

BlackMan_阿伟

最新推荐文章于 2024-09-01 16:56:02 发布

阅读量2.8k

点赞数 4

分类专栏：深度学习环境文章标签： linux python docker

本文链接：https://blog.csdn.net/weixin_37724529/article/details/107676543

版权

情景：加入实验室有8块卡，前7块卡已经呗占用了，如果我们需要运行程序，肯定会错，错误如下：

Traceback (most recent call last):
  File "train.py", line 182, in <module>
    fasterRCNN = vgg16(classes, pretrained=True, class_agnostic=args.class_agnostic)
  File "/home/lthpc/lw/WeaklyObjectDetection/lib/model/faster_rcnn/vgg16.py", line 26, in __init__
    _fasterRCNN.__init__(self, classes, class_agnostic)
  File "/home/lthpc/lw/WeaklyObjectDetection/lib/model/faster_rcnn/faster_rcnn.py", line 35, in __init__
    self.RCNN_rpn = _RPN(self.dout_base_model)
  File "/home/lthpc/lw/WeaklyObjectDetection/lib/model/rpn/rpn.py", line 74, in __init__
    sequence_length=sequence_length)
  File "/home/lthpc/lw/WeaklyObjectDetection/lib/model/rpn/bi_lstm.py", line 50, in __init__
    self.u_omega = Variable(torch.randn(self.attention_size).cuda(), requires_grad=True)
RuntimeError: CUDA error: out of memory