Загрузка модели torchscript в C ++: невозможно инициализировать CUDA без библиотеки ATen_cuda

Question

Я построил CNN с Pytorch с поддержкой GPU. Он отлично работает в Python, и я хотел бы импортировать его в моем приложении C ++, используя VS2017. Я следую этим инструкциям здесь и здесь .
Я могу собрать и запустить приложение, но я получаю эту ошибку при загрузке модели:

error loading the model
Cannot initialize CUDA without ATen_cuda library. PyTorch splits its backend into two shared 
libraries: a CPU library and a CUDA library; this error has occurred because you are trying to use 
some CUDA functionality, but the CUDA library has not been loaded by the dynamic linker for some 
reason.  The CUDA library MUST be loaded, EVEN IF you don't directly use any symbols from the CUDA 
library! One common culprit is a lack of -Wl,--no-as-needed in your link arguments; many dynamic 
linkers will delete dynamic library dependencies if you don't depend on any of their symbols.  You 
can check if this has occurred by using ldd on your binary to see if there is a dependency on 
*_cuda.so library. (initCUDA at 
C:\w\1\s\windows\pytorch\aten\src\ATen/detail/CUDAHooksInterface.h:63)
(no backtrace available)

Я не могу найти ATen_cuda file в C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1.
У кого-нибудь есть Windows C ++ VS2017 Pytorch с графическим процессором?

Answer 1

Я решил использовать ночную сборку , которая поставляется pytorch со всеми необходимыми cuda библиотеками.
Похоже, что установщик conda для Windows не устанавливаетпоследний pytorch stable 1.2, но более старый 1.0.1. Ночная сборка для Windows Conda Python 3.7 устанавливает более новую 1.3.