본문 바로가기
DL|ML

multi gpu 사용 가능 환경에서 cpu 사용하기

by 이든Eden 2023. 6. 10.

회사 개발 서버에서 multi gpu를 사용하는데 gpu가 모자라는 상황이었다. (자주 있음..)

어쨌든 이런 상황에서 cpu로 돌리려면 아래와 같이 하면 된다.

 

CUDA_VISIBLE_DEVICES=-1 python filename.py