[AI-Toolkit]blackwell 그래픽카드 대응 세팅

Stability Matrix를 사용하면 AI-Toolkit이나 Comfyui 설치가 용이하다.

AI-Toolkit을 사용할 때 Blackwell 그래픽카드는 몇 가지 세팅을 바꿔야한다.

  1. numpy==1.26.4
  2. scipy==1.13.1
  3. torch는 nightly로 설치
    • 설치 전 기존 토치를 깔끔하게 삭제
      • python -m pip uninstall -y torch torchvision torchaudio
      • python -m pip cache purge
    • nightly로 토치 설치
    • 원래 오디오 토치는 함께 설치 되나 혹시 오디오 토치가 설치되지 않을 시
      • pip install –pre torchaudio –index-url https://download.pytorch.org/whl/nightly/cu130

코멘트

답글 남기기

이메일 주소는 공개되지 않습니다. 필수 필드는 *로 표시됩니다