본문 바로가기
  • AI 개발자가 될래요
Python

[PyTorch] CUDA capability sm_86 is not compatible with the current PyTorch installation

by 꿀개 2022. 12. 12.

실험도 여러 개 하고.. 장비도 여러 개 사용하다보니 환경 문제에 자주 부딪힌다.

오늘은 DeepFake 오픈소스 좀 돌려보려 했더니 

 

이런 에러가 떴다.

 

오픈소스 requirements에는 pytorch 1.8.1, cuda 10.2 를 사용해서

# CUDA 10.2
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch

이 명령어로 설치했었는데.. 이 버전이랑 RTX A5000이랑 안맞나보다ㅜ

 

결국 노가다로 맞는 버전 찾을 수 밖에ㅡㅜㅠ

 

아래 명령어로 pytorch랑 cudatoolkit을 지우고 다시 깔아보자. 

conda uninstall pytorch cudatoolkit

 

  • PyTorch v1.12.1 cuda 11.3
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

오잉! 혹시나 하고 비교적 최신 버전으로 설치해봤는데 실행된다. 

 

환경 문제는 답이 명확하게 없는 것 같다.. 그냥 여기선 되고 저기선 안되고 알아서 방법을 찾아야할듯 ㅜㅜ