pytorch: gpu 선택해서 사용하는 방법

djlee·2023년 3월 13일

0

다음과 같이 gpu 번호를 설정하면 돼요.

select_gpu = 1
torch.cuda.set_device(select_gpu)

이전 포스트

가상 환경: requirements.txt를 통해 가상 환경 설치하는 방법

다음 포스트

pytorch: multi-gpu 사용 시 attribute 접근 방법

0개의 댓글