LibTorch使用cuda的问题

[wang8713]

你好，我刚刚接触libtorch但在使用的过程中出现了问题。我使用gpu的时候在运行 "module.to(device_type);" 时出现了不动的问题。 情况如下图。在运行到81行给模型加载到gpu时出现了下面这种情况。 在使用cpu时一切正常。 我的环境是 cuda10.2、cudnn(8.3.1 / 7.6.5)这两个版本都试过、libtorch1.8.0-release、VS2019、pytorch1.8.0。

[AllentDan]

点击这里的一些报错部分

[wang8713]

好的，谢谢您回复，我已经解决了。