Resultados da pesquisa a pedido "caffe2"

2 a resposta

Pode Caffe ou Caffe2 receber dados de entrada diretamente da GPU?

Eu li os tutoriais do caffe2 e tentei modelos pré-treinados. Eu sabia que o caffe2 utilizaria a GPU para executar o modelo / net. Mas os dados de entrada sempre parecem ser dados da memória da CPU (ou seja, host). Por exemplo, em Carregando ...