[pytorch] jupyter notebook gpu지정

ma-kjh·2024년 3월 11일

0

Pytorch

목록 보기

16/20

os.environ["CUDA_VISIBLE_DEVICES"]= 사용하고자 하는 GPU

거인의 어깨에 올라서서 더 넓은 세상을 바라보라 - 아이작 뉴턴

이전 포스트

[pytorch]nn.BatchNorm2d()

다음 포스트

[conda] create env

0개의 댓글