Última pergunta sobre Libtorch

0
Responda

Leitura do array cuda no torch c++ sem transferir o array para a CPU

Estou tentando ler o array cuda localizado na GPU no torch c++ cuda. Embora eu tenha lido a matriz, mas parece que a matriz está primeiro transferindo a matriz cuda para a CPU e, novamente, em tocha, está transferindo a matriz para cuda em sua estrutura. Posso dizer isso porque o uso da CPU é bastante alto e a transferência da matriz para a tocha não é tão suave. Estou usando o seguinte código, que está funcionando.

c++ python 3.x pytorch libtorch torch

0
Responda

Como usar torch::get_hash na libtorch?

Tenho tentado obter o hash de alguns módulos na libtorch sem sucesso até agora. Na libtorch, existem várias funções relacionadas ao hashing, como: get_hash e simple_hash_get que são definidos no torch/csrc/utils/hash.h.

libtorch c++

Etiquetas quentes

© 2021   OlaMundo.Org