본문 바로가기

cuDNN2

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.
Nvidia 그래픽 드라이버/CUDA/cudnn 설치 및 제거 Ubuntu 20.04 기준 Nvidia 그래픽 드라이버 삭제 sudo apt-get purge nvidia* sudo apt-get autoremove sudo apt-get autoclean CUDA 삭제 sudo rm -rf /usr/local/cuda* sudo apt-get --purge remove 'cuda*' sudo apt-get autoremove --purge 'cuda*' 제거 확인 #그래픽 드라이버 sudo dpkg -l|grep nvidia # CUDA sudo dpkg -l|grep cuda # 만약 설치된게 남았다면: sudo apt-get remove --purge 남은 찌꺼기 이름 이후 재부팅 Nvidia 그래픽 드라이버 설치 ubuntu-drivers devices 를 .. 2022. 11. 14.

티스토리툴바