반응형
안녕하세요~
저의 경우 GPU가 4개 달린 환경에서 학습을 진행하고 있습니다~
학습 코드에서 GPU할당을 자동으로 나눠서 할당 하게끔 짜여있지만 종종 하나의 GPU에서만 동작 시키고 싶을 때가 있을겁니다.
그럴 경우 CUDA_VISIBLE_DEVICES = x 를 사용하시면 됩니다.
예를 들어 4개의 GPU중 4번 째 GPU만 사용하고 싶다! 그러면
CUDA_VISIBLE_DEVICES=3 python3 ~~다음과 같이 사용하시면 됩니다~
읽어주셔서 감사합니다
반응형
'python' 카테고리의 다른 글
| [python] class 오버라이딩(overriding), 상속, super() (0) | 2021.11.09 |
|---|---|
| DataLoader worker is killed by signal: Killed 에러 (0) | 2021.11.04 |
| [python]확장자 일괄 변경 (0) | 2021.11.03 |
| json to xml convert 작업[ 작성중..] (0) | 2021.10.22 |
| [python] list함수 (append, insert, pop, remove, sort) (0) | 2021.06.29 |