я пытаюсь запустить код Python, где код запускает библиотеку face_recognition из
, где требуется графический процессор для более быстрой работы для dlib
модели запуска face_recognition CNN
я использую облачное изображение NVIDIA GPUдля Deep Learning и HPC для запуска сценария python, затем сценарий python main.py
:
from PIL import Image
import face_recognition
from datetime import datetime
start=datetime.now()
# Load the jpg file into a numpy array
image = face_recognition.load_image_file("com.jpg")
face_locations = face_recognition.face_locations(image, number_of_times_to_upsample=0, model="cnn")
face_encode = face_recognition.face_encodings(image, face_locations)
print("I found {} face(s) in this photograph.".format(len(face_locations)))
print(face_encode)
print(datetime.now()-start)
здесь драйвер NVIDIA из nvidia-smi
версия драйвера 396.44: драйвер NVIDIA
Я установил CUDA-Toolkit 9.2 и CUDNN, вот журнал при установке dlib
с поддержкой GPU:
там есть поддержка CUDA будет использоваться для dlib
, затем я пытаюсь запуститькод main.py
, но процесс, не запущенный на графическом процессоре NVIDIA, занял 01: 06.114585
есть идеи для этой проблемы?цель в том, чтобы я просто хотел запустить main.py
процесс на ядре графического процессора. Спасибо