我正在尝试运行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
进程,谢谢