当前位置: 代码迷 >> 综合 >> libtorch1.6 GPU版本无法使用的问题
  详细解决方案

libtorch1.6 GPU版本无法使用的问题

热度:75   发布时间:2023-12-08 17:07:11.0

    在官网中下载libtorch1.6 GPU版本,如果在VS中只添加inclue和lib的路径,编译的过程或编译好执行的过程很有可能出现错误。当然,如果是使用cmake来构建工程,是可以正常编译和执行。

    出现错误的解决方法:

        1)使用VS2017及以上版本;

        2)windows上装的cuda版本需要与下载的libtorch的cuda版本相对应;

        3)在“属性 --> 链接器 --> 命令行 --> 其他选项”中添加:

  /INCLUDE:?warp_size@cuda@at@@YAHXZ