• 카테고리

    질문 & 답변
  • 세부 분야

    게임 프로그래밍

  • 해결 여부

    미해결

GPU로 트레이닝 시키는 방법

22.04.07 22:07 작성 조회수 120

1

아나콘다로 트레이닝을 시킬때 CPU로 계속 실행이 되는데요, GPU로 실행시키는 방법은 없나요? 감사합니다

답변 2

·

답변을 작성해보세요.

2

일반적으로 mlagents 관련 패키지를 설치하시면 기본적으로 cpu 버전으로 설치되는 것으로 알고있습니다. cuda를 설치하시고 해당 cuda 버전에 맞는 pytorch gpu 버전을 새로 설치해주시면 됩니다! 

1

고재청님의 프로필

고재청

질문자

2022.04.12

네 해결됐습니다. 감사합니다!