我该怎么使用gpu调用动物识别模型

[Nkomol]

查看了很多官方文档，发现都是使用cpu调用的，GPU调用aistudio中也没有找到，请问有GPU调用的环境安装教程以及使用gpu调用该模型的方式吗？抱歉可能因为我是一个初学者才会问出这个问题。

[jm12138]

不同模型调用 GPU 的方式可能不太一样，具体得看文档，动物识别模型的话只需要设置环境变量 CUDA_VISIBLE_DEVICES=0 即可调用设备中的第一张显卡。

[Nkomol]

我设置的环境变量，同时也配置完成了gpu环境但是使用这个识别模型时resnet50_vd_animals错误提示：AttributeError: 'ResNet50vdAnimals' object has no attribute 'gpu_predictor' 已设置环境变量，且设置默认GPU模型，使用其他模型GPU是正常的代码如下： paddle.device.set_device('gpu:0')

使用 resnet50_vd_animals 模型进行检测

results = module.classification(images=[np_images],use_gpu=True)

[jm12138]

• 可以试试使用 1.1.0 版本的 resnet50_vd_animals。

• 使用下面的代码测试，应该是可以正常使用的。

  import os
  import cv2
  import paddlehub as hub
  
  os.environ['CUDA_VISIBLE_DEVICES'] = '0'
  
  classifier = hub.Module(name="resnet50_vd_animals")
  
  result = classifier.classification(images=[cv2.imread('/PATH/TO/IMAGE')])

[Nkomol]

使用了这个代码，启动时是有看到gpu跳动的，但是在识别的过程中，gup的占用一直为0，是因为这么模型使用时基本不暂用吗？

[jm12138]

使用了这个代码，启动时是有看到gpu跳动的，但是在识别的过程中，gup的占用一直为0，是因为这么模型使用时基本不暂用吗？

按道理会有1g左右的显存占有，GPU利用率的话，这个模型挺小的，没啥波动也正常