[일반] 로컬 gpu로 돌리는 거 질문좀

[익명(121.143)]

torch.cuda.is_available() 이거 True 띄우는데 성공은 했는데

그러면 앞으로 tensorflow, torch 이런 거 쓸 때 자동으로 cpu 대신 글카 쓰는 거임?

아니면 코드 돌리기 전에 뭔가 선언해줘야함?

[1(14.37)]

모델이나 변수 선언 한 다음에 model.to("cuda:0") input.to("cuda:0")이런식으로 GPU로 보내줘야함

[익명(121.143)]

진자 댕큐 합니다

[익명(121.176)]

아님 뭐 to device 뭐시기 하던가 - dc App

[Cytrus(minhyoo1)]

device 정의하고, to device. - dc App