libtorch 環境搭建

1.下載libtorch
(1)Pytorch 下載官網：https://pytorch.org/get-started/locally/
(2)選擇版本為最新的1.3版,操作系統為windows,包為libtorch,語言為c++,cuda 版本為最新的10.1。如下圖：
在這里插入圖片描述
(3)點擊release 版的libtorch 庫的下載鏈接，下載壓縮包。
2.解壓下載到的壓縮文件到任意目錄
3.下載安裝CUDA 10.1
(1)需要英偉達的支持cuda的顯卡，我的顯卡是GTX950M。如下圖：

4.測試代碼如下：
#include
#include <torch/torch.h>
using namespace std;

int main()
{
torch::Tensor tensor = torch::rand({ 2, 3 });

std::cout << tensor.cuda() << std::endl;

}
5.Vs2019 工程環境配置：
(1)選擇release x64
(2)Include 目錄：
在這里插入圖片描述
(3)Lib 目錄:

(4)Lib庫：

6.結果

本文鏈接：https://blog.csdn.net/mceet/article/details/103433953

智能推薦

requests實現全自動PPT模板

http://www.1ppt.com/moban/ 可以免費的下載PPT模板，當然如果要人工一個個下，還是挺麻煩的，我們可以利用requests輕松下載訪問這個主頁，我們可以看到下面的樣式點每一個PPT模板的圖片，我們可以進入到詳細的信息頁面，翻到下面，我們可以看到對應的下載地址點擊這個下載的按鈕，我們便可以下載對應的PPT壓縮包那我們就開始做吧首先，查看網頁的源代碼，我們可以看到每一...

SpringMvc自定義類型轉換器步驟

...

Linux C系統編程-線程互斥鎖（四）

互斥鎖互斥鎖也是屬于線程之間處理同步互斥方式，有上鎖/解鎖兩種狀態。互斥鎖函數接口 1）初始化互斥鎖 pthread_mutex_init() man 3 pthread_mutex_init (找不到的情況下首先 sudo apt-get install glibc-doc sudo apt-get install manpages-posix-dev) 動態初始化 int pthread_...

統計學習方法 - 樸素貝葉斯

引入問題：一機器在良好狀態生產合格產品幾率是 90%，在故障狀態生產合格產品幾率是 30%，機器良好的概率是 75%。若一日第一件產品是合格品，那么此日機器良好的概率是多少。貝葉斯模型生成模型與判別模型判別模型，即要判斷這個東西到底是哪一類，也就是要求y，那就用給定的x去預測。生成模型，是要生成一個模型，那就是誰根據什么生成了模型，誰就是類別y，根據的內容就是x 以上述例子，判斷一個生產出...

styled-components —— React 中的 CSS 最佳實踐

https://zhuanlan.zhihu.com/p/29344146 Styled-components 是目前 React 樣式方案中最受關注的一種，它既具備了 css-in-js 的模塊化與參數化優點，又完全使用CSS的書寫習慣，不會引起額外的學習成本。本文是 styled-components 作者之一 Max Stoiber 所寫，首先總結了前端組件化樣式中的最佳實踐原則，然后在此基...

libtorch 環境搭建

智能推薦

猜你喜歡