Skip to content

Commit

Permalink
Removed makefile; updated README
Browse files Browse the repository at this point in the history
  • Loading branch information
petertorelli committed Jan 21, 2020
1 parent 791e02e commit 692afbd
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 674 deletions.
15 changes: 9 additions & 6 deletions targets/tensorrt/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -85,21 +85,24 @@ These folders are created after the first run:

`Results`: Interim results in the form of text files are stored in this folder.

## Compiling
## Compiling & Dependencies

As stated earlier, the default libraries were compiled on Arm architecture for both Nano and Xavier. The libraries can be recompiled for x86_64 as well by doing the following:

In order to compile, you must checkout NVlabs [cub](https://github.com/NVlabs/cub) and then add this line to `cpp_environment/CMakeLists.txt`:

~~~
include_directories(/home/dev/cub)
~~~

...where `/home/dev/cub` is the path to the repository. Then you can build:

~~~
% cd targets/tensorrt/cpp_environment
% cmake .
% make
~~~

### Dependencies
* CUB: https://github.com/NVlabs/cub, download and extract then change the include_directories in cpp_environment/CMakeLists.txt

include_directories(/home/dev/cub-1.8.0)

## Known Issues:

* Sometimes process may get "killed" while creating an engine. This may be due to insufficient system resources, so make sure no other applications are running. Simply re-running should solve the problem, as all models have been tested for batch sizes up to 16 with success.
Expand Down

0 comments on commit 692afbd

Please sign in to comment.