Build fixes for ubuntu 20.04 (triton-inference-server#7)

Co-authored-by: David Goodwin <[email protected]>

Author: dzier

CMakeLists.txt

@@ -44,6 +44,9 @@ if(NOT CMAKE_BUILD_TYPE)
   set(CMAKE_BUILD_TYPE Release)
 endif()
 
+# Python.h needed by torch headers.
+find_package(Python3 REQUIRED COMPONENTS Development)
+
 #
 # Dependencies
 #
@@ -100,6 +103,7 @@ target_include_directories(
   PRIVATE
     ${CMAKE_CURRENT_SOURCE_DIR}/src
     ${TRITON_PYTORCH_INCLUDE_PATHS}
+    ${Python3_INCLUDE_DIRS}
 )
 
 target_compile_features(triton-pytorch-backend PRIVATE cxx_std_11)

README.md

@@ -52,7 +52,7 @@ main Triton [issues page](https://github.com/triton-inference-server/server/issu
 Use a recent cmake to build. First install the required dependencies.
 
 ```
-$ apt-get install patchelf rapidjson-dev python3.6-dev
+$ apt-get install patchelf rapidjson-dev python3-dev
 ```
 
 Copy over the LibTorch and Torchvision headers and libraries from the
@@ -62,7 +62,7 @@ into local directories.
 ```
 $ mkdir build
 $ cd build
-$ cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install -DTRITON_PYTORCH_INCLUDE_PATHS="/opt/tritonserver/include/torch;/opt/tritonserver/include/torch/torch/csrc/api/include;/opt/tritonserver/include/torchvision;/usr/include/python3.6" -DTRITON_PYTORCH_LIB_PATHS="/opt/tritonserver/backends/pytorch"..
+$ cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install -DTRITON_PYTORCH_INCLUDE_PATHS="/opt/tritonserver/include/torch;/opt/tritonserver/include/torch/torch/csrc/api/include;/opt/tritonserver/include/torchvision" -DTRITON_PYTORCH_LIB_PATHS="/opt/tritonserver/backends/pytorch"..
 $ make install
 ```