DL|ML

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

이든Eden 2023. 6. 10. 19:58

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

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

 

CUDA_VISIBLE_DEVICES=-1 python filename.py