libtorch gpu vs2017環(huán)境配置

一、下載

https://pytorch.org/
下載release版本并解壓（參考網(wǎng)上博客）

二、vs2017配置

  1. 右擊解決方案-->屬性-->C/C++-->常規(guī)-->附加包含目錄

    如圖，添加libtorch下的兩個文件夾

  2. 右擊解決方案-->屬性-->鏈接器-->常規(guī)-->附加目錄項

    添加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");
}

如果結(jié)果為1，則環(huán)境配置成功

四、新坑記錄

符合模式選否（默認(rèn)是）


最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容