我用带有GPU支持的Pytorch构建了CNN。它在python中工作正常,我想使用VS2017将其导入我的C ++应用程序中。我正在按照here和here这些说明进行操作。
我可以构建和运行该应用程序,但是在加载模型时出现此错误:
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且具有GPU工作示例吗?