无法在NVIDIA GPU Cloud Image上使用GPU在深度学习和HPC上运行Process dlib

时间:2018-09-24 09:53:27

标签: python nvidia face-recognition

我正在尝试运行python代码,其中代码从face_recognition运行face_recognition库,而dlib face_recognition运行模型CNN需要GPU更快地运行

我正在使用NVIDIA GPU Cloud Image进行深度学习和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-smi版驱动程序版本396.44中的NVIDIA驱动程序: NVIDIA Driver

我已经安装了CUDA-Toolkit 9.2和CUDNN,这里是安装dlib并支持GPU的日志: Log有支持CUDA用于dlib

然后我尝试运行代码main.py,但是未在NVIDIA GPU上运行的过程花费了01:06.114585 run code这个问题的任何想法?目标是我只想在GPU内核上运行main.py进程,谢谢

0 个答案:

没有答案