Study/시행착오
[해결] CUDA capability sm_86 is not compatible with the current PyTorch installation.
- -
728x90
원격 서버에서 모델을 테스트 겸 돌려보려다 발생한 에러입니다. pytorch와 CUDA가 충돌을 일으킨다고 합니다. 제가 이해가 안 되는 점은.. 지난주에는 멀쩡하게 잘만 되던 cuda가 오늘 말썽인 건지... 아마 이번 모델을 돌리기 전에 필요한 패키지들을 설치하는 중간에 pytorch가 재설치되는 과정에서 뭔가 꼬인 듯합니다.
pytorch를 삭제하고 운영체제와 쓰려는 CUDA 버전에 맞추어 재설치하니까 해결이 되긴 했습니다.
# 파이토치 uninstall
conda uninstall pytorch
# linux, cuda 11.8 설치
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia아래 홈페이지에서 각자 사용 환경에 맞추어 재설치를 시도해보시는 것을 추천드려요.
728x90
'Study > 시행착오' 카테고리의 다른 글
| [해결] Watchdog caught collective operation timeout, Multi-GPU (DDP) NCCL 타임아웃 시간 설정 (0) | 2023.06.23 |
|---|---|
| [ongoing] 깃허브 키체인 항목 접근 권한 변경에 대한 암호 입력 (1) | 2023.05.08 |
| [해결] 원격 서버에서 도커 실행이 안될 때 (0) | 2023.03.08 |
| [on-going] AssertionError: can only test a child process (0) | 2022.11.23 |
| [해결] black 사용 시 cannot import name 'ParameterSource' from 'click.core' (0) | 2022.09.06 |
Contents
공감 감사합니다! 좋은하루 되세요 ✨