TRtorch 엔진으로 pytorch 모델을 컴파일하려고 합니다. 이 링크 에 따라 TRtorch를 설치했습니다 .
샘플 코드가 실행될 때(이 링크 의 아래 명령 사용 ) 주어진 오류가 발생합니다.
sudo bazel run //cpp/trtorchexec -- $(realpath /home/TRTorch/tests/modules/alexnet_scripted.jit.pt) "(1,3,227,227)"
공유 라이브러리 로드 중 오류: libnvinfer.so.7: 공유 개체 파일을 열 수 없음: 해당 파일 또는 디렉터리가 없습니다.
또한 LD_LIBRARY_PATH 가 올바르게 설정되었습니다.
내보내기 LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/TensorRT/TensorRT-7.0.0.11/lib
더 많은 정보:
TRtorch: 최신 버전(python 패키지 및 바이너리)
TensorRT: 7.0.0.11
Pytorch: 1.5.1
CUDA: 10.2
Python: 3.6
이 기사는 인터넷에서 수집됩니다. 재 인쇄 할 때 출처를 알려주십시오.
침해가 발생한 경우 연락 주시기 바랍니다[email protected] 삭제
몇 마디 만하겠습니다