libtorch如何使用gpu?

问题遇到的现象和发生背景

我的电脑cuda是11.3,cudnn也是对应下好的,libtorch
下载的是1.10.0的版本(对应的cu113)release64

用代码块功能插入代码,请勿粘贴截图

我的代码只能运行cpu,无法运行gpu,尝试过各种办法都无法解决,包括在命令行加上/INCLUDE:?warp_size@cuda@at@@YAHXZ

img

就会显示无法解析。不加上可以正常运行但是无法启用gpu

img

我的解答思路和尝试过的方法

我的电脑cuda113,libtorch也是对应的版本releas

img

我想要达到的结果

能够启用gpu

不知道

参考:

或者自己在本地重新编译libtorch