Ubuntu, torch-tensorrt 설치 및 사용
Ubuntu 20.04 Nvidia graphic driver version: 515.65.01 CUDA version: 11.6 Cudnn version: 8.4.1 1. 주의할점 pip 명령줄 한줄이면 설치가 되나, CUDA, cudnn, torch, torchivision, torchaudio, 그리고 마지막 torch-tensorrt 의 버전이 모두 맞아야 작동한다. 또한, 설치가 꼬였을때, torch, torchvision, torchaudio, torch-tensorrt 모두 지웠다가 새로 설치를 해야한다. (CUDA, cudnn은 그대로 둬도 된다) 2. torch 설치 현재 torch 1.13 버전이 배포되고 있으나, 에러가 발생하여 1.12.1 버전으로 진행 공식 링크(Previous P..
2022. 11. 27.
