Inflearn Community Q&A
asked
GPU로 트레이닝 시키는 방법
Written on
·
299
1
아나콘다로 트레이닝을 시킬때 CPU로 계속 실행이 되는데요, GPU로 실행시키는 방법은 없나요? 감사합니다
unity-ml-agents머신러닝 배워볼래요? unity강화학습
Answer 2
2
kyushik
Instructor
일반적으로 mlagents 관련 패키지를 설치하시면 기본적으로 cpu 버전으로 설치되는 것으로 알고있습니다. cuda를 설치하시고 해당 cuda 버전에 맞는 pytorch gpu 버전을 새로 설치해주시면 됩니다!
1
