[pytorch] jupyter notebook gpu지정

-·2024년 3월 11일

0

Pytorch

목록 보기

16/26

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

살아남은 자가 강한 것

이전 포스트

[pytorch]nn.BatchNorm2d()

다음 포스트

[conda] create env

0개의 댓글