~/temp/server$ cat Dockerfile
FROM nvcr.io/nvidia/tritonserver:22.05-py3
RUN nvcc -V
WORKDIR /data
COPY ./model_repository /data/model_repository
WORKDIR /data/model_repository
RUN /usr/src/tensorrt/bin/trtexec --onnx=./helloworld/1/model.onnx --saveEngine='./helloworld/1/model.plan' --workspace=1024 --minShapes=input1:1x3x640x640 --optShapes=input1:2x3x640x640 --maxShapes=input1:4x3x640x640
RUN rm -rf ./*/1/model.onnx
RUN apt-get update
RUN apt-get install ffmpeg libsm6 libxext6 -y
RUN pip install opencv-python
RUN pip install scikit-image
Thursday, 28 July 2022
how to make a tritonserver dockerfile?
Wednesday, 27 July 2022
docker compose trtexec returned a non-zero code: 139, Code: 139 nvidia triton server docker
peseyes@peseyes-police-logistic:~/temp$ cat /etc/docker/daemon.json
{
"runtimes": {
"nvidia": {
"path": "nvidia-container-runtime",
"runtimeArgs": []
}
},
"default-runtime": "nvidia"
}
Thursday, 14 July 2022
Thursday, 7 July 2022
Thursday, 23 June 2022
how to write a unit test using googletest in ubuntu?
(base) ninja@luke:~/workspace/google_test$ cat main_test.cpp
// main_test.cpp
#include "whattotest.cpp"
#include <gtest/gtest.h>
TEST(SquareRootTest, PositiveNos) {
ASSERT_EQ(6, squareRoot(36.0));
ASSERT_EQ(18.0, squareRoot(324.0));
ASSERT_EQ(25.4, squareRoot(645.16));
ASSERT_EQ(0, squareRoot(0.0));
}
TEST(SquareRootTest, NegativeNos) {
ASSERT_EQ(-1.0, squareRoot(-15.0));
ASSERT_EQ(-1.0, squareRoot(-0.2));
}
int main(int argc, char **argv) {
testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}
---
(base) ninja@luke:~/workspace/google_test$ cat whattotest.cpp
// whattotest.cpp
#include <math.h>
double squareRoot(const double a) {
double b = sqrt(a);
if(b != b) { // nan check
return -1.0;
}else{
return sqrt(a);
}
}
---
(base) ninja@luke:~/workspace/google_test$ cat CMakeLists.txt
cmake_minimum_required(VERSION 3.16)
project(test)
################################
# GTest
################################
project(googletest-git NONE)
include(FetchContent)
FetchContent_Declare(
googletest
GIT_REPOSITORY https://github.com/google/googletest.git
GIT_TAG release-1.8.0
)
set(gtest_force_shared_crt ON CACHE BOOL "" FORCE)
set(BUILD_GMOCK OFF CACHE BOOL "" FORCE)
set(BUILD_GTEST ON CACHE BOOL "" FORCE)
FetchContent_MakeAvailable(googletest)
################################
# main test program
################################
# Add test cpp file
add_executable(test main_test.cpp)
# Link test executable against gtest & gtest_main
target_link_libraries(test gtest gtest_main)
commands:-
cd ~/workspace/google_test
mkdir build
cd build
cmake ..
make test
How to include all *.h or *.cpp in CMakeLists.txt?
file(GLOB_RECURSE SOURCES LIST_DIRECTORIES true *.h *.cpp)
How to create multiple folders in one go using terminal?
mkdir -p test/{build,lib,src,tst}