반응형 Deep Learning (Computer Vision)/Model Compression and Optimization3 TensorRT설치하기 우선 Python 3.10.12 기준으로 설치할 예정이며, CUDA 버전은 nvcc -V를 했을 때 아래와 같다. 혹시 모르니 nvidia-smi를 했을 때의 환경 또한 첨부하자면 아래와 같다. 근데 지금 확인했는데 왜.. nvidia-smi에 12.1로 잡히는거지...어쨌든 11.8 CUDA를 기준으로 설치했다. 파이토치에서 TensorRT를 그냥 대충 설치하고 실행하면 되겠지 하고 안일한 마음으로 실행하다가 3일을 날려먹은 나레기 혹시 나와 같은 실수를 하는 사람이 있을지 모르기 때문에, 일단 기록용으로 적어두려고 한다. RuntimeError: forward() expected at most 1 argument(s) but received 2 argument(s). Declaration: forwa.. 2023. 8. 31. Pytorch Tutorial보다 친절한 Pytorch Pruning Tutorial (2) Pytorch Pruning 1편 보러가기. Pytorch Tutorial보다 친절한 Pytorch Pruning Tutorial (1) Colab 자료 - https://colab.research.google.com/github/pytorch/tutorials/blob/gh-pages/_downloads/f40ae04715cdb214ecba048c12f8dddf/pruning_tutorial.ipynb#scrollTo=mRMctJEUvqbS 번역본 - https://tutorials.pytorch.kr/intermediate/pruning_tutorial.html - 187cm.tistory.com 이번시간에는 지난시간에 이어서 Pytorch Pruning Tutroial을 이어서 진행해보려 한다. I.. 2023. 6. 1. Pytorch Tutorial보다 친절한 Pytorch Pruning Tutorial (1) Colab 자료 - https://colab.research.google.com/github/pytorch/tutorials/blob/gh-pages/_downloads/f40ae04715cdb214ecba048c12f8dddf/pruning_tutorial.ipynb#scrollTo=mRMctJEUvqbS 번역본 - https://tutorials.pytorch.kr/intermediate/pruning_tutorial.html - 위에 존재하는 번역본을 조금 더 자세하게 풀어써보며, Pruning을 공부해보자 - 위에 내용이 보고 따라하면 충분할 순 있지만, 왜 이게 이렇게 될까를 생각해보면 약간 불친절한 설명서이다. 이부분을 조금 더 친절하게 설명해보고자 한다. 우선 %matplotlib inlin.. 2023. 5. 29. 이전 1 다음 반응형
