libtorch gpu vs2017环境配置

一、下载

https://pytorch.org/
下载release版本并解压（参考网上博客）

二、vs2017配置

  1. 右击解决方案-->属性-->C/C++-->常规-->附加包含目录

    如图，添加libtorch下的两个文件夹

  2. 右击解决方案-->属性-->链接器-->常规-->附加目录项

    添加lib路径

  3. 右击解决方案-->属性-->链接器-->输入-->附加依赖项


    c10.lib
    c10_cuda.lib
    torch_cpu.lib
    torch_cuda.lib
    torch_cuda_cu.lib
    torch_cuda_cpp.lib
    -INCLUDE:?warp_size@cuda@at@@YAHXZ
    最后一项十分重要
    4.右击解决方案-->属性-->链接器-->输入-->命令行


三、测试

#include<iostream>
#include<torch/script.h>
#include <torch/torch.h> 
#include<memory>
 
int main()
{
    std::cout <<"cuda::is_available():" << torch::cuda::is_available() << std::endl;
    system("pause");
}

如果结果为1，则环境配置成功

四、新坑记录

符合模式选否（默认是）


最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容