인프런 커뮤니티 질문&답변

고재청님의 프로필 이미지

작성한 질문수

유니티 머신러닝 에이전트 완전정복 (기초편)

Behavioral Cloning 이론

GPU로 트레이닝 시키는 방법

22.04.07 22:07 작성

·

216

1

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

답변 2

2

민규식님의 프로필 이미지
민규식
지식공유자

2022. 04. 07. 22:24

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

1

고재청님의 프로필 이미지
고재청
질문자

2022. 04. 12. 12:53

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