linux Pytorch C++ 环境搭建

 

Pytorch C++ 环境搭建

0，阅读Pytorch C++官网文档，在虚拟机上安装Ubuntu16.04的Linux操作系统搭建环境，并使用makefile进行编译，代码简洁易读。

sudo apt-get update

sudo apt-get install make cmake libnss3 tree

 

1，下载pytorch c++ cpu 库文件，并解压，注意此目录路径，cmake编译需要

unzip libtorch-shared-with-deps-latest.zip

pwd

 

2，安装vscode，下载地址：linux vscode，解压并启动

sudo dpkg -i code_1.32.1-1552006243_amd64.deb

code

 

3，创建一个工程，目录结构：

rose@rose-machine:~/pytorch_c++/test$ tree

1. .

2. ├── main.cpp # 主程序

3. ├── bin # cmake 编译输出的文件目录

4. ├── CMakeLists.txt # cmake文件

5. └── makefile # make编译CMakeLists.txt，代码简洁

main.cpp，代码

1. #include

2. #include

3.  

4. #include

5. #include

6.  

7. using namespace std;

8. using namespace at;

9.  

10. int main(int argc, const char* argv[])

11. {

12. at::Tensor a = at::ones({2,2}, at::kInt);

13. std::cout<< a << "\n";

14.  

15. std::cout<< "ok\n";

16. return 1;

17. }

CMakeLists.txt，代码

1. cmake_minimum_required(VERSION 3.0 FATAL_ERROR)

2.  

3. # 项目名称

4. project(demo)

5.  

6. find_package(Torch REQUIRED)

7.  

8. # 输出项目名称，依赖main.cpp

9. add_executable(${PROJECT_NAME} main.cpp)

10. target_link_libraries(${PROJECT_NAME} "${TORCH_LIBRARIES}")

11. set_property(TARGET ${PROJECT_NAME} PROPERTY CXX_STANDARD 11)

makefile，代码

1. .PHONY:all clean

2.  

3. # bin为cmake输出目录

4. PATH_BIN=bin

5. # pytorch_lib为libtorch-shared-with-deps-latest.zip解压后的目录

6. pytorch_lib=/home/rose/software/libtorch

7.  

8. all:

9. @cd ./$(PATH_BIN) && cmake -DCMAKE_PREFIX_PATH=$(pytorch_lib) .. && make

10.  

11. clean:

12. rm -rf $(PATH_BIN)

13. mkdir $(PATH_BIN)

 

4，编译与执行。

4.1，编译，make

4.2，运行，./bin/demo

4.3，删除cmake编译文件，make clean

 

5，下载此项目，test.tar.gz