Q. GPU 달린 내 컴퓨터에 제대로 pytorch가 설치되었는지 테스트 하는 방법은?

A. python을 실행한다.

아래처럼 입력한다.

$ python
import torch
torch.rand(5).to('cuda')

 

제대로 작동한다면 아래처럼 나온다.

cuda를 써서 난수를 얻어낸 것이다.

 

torch.rand(5) 는 잘 실행되는데,

torch.rand(5).to('cuda')는 엄청 오래 걸리고 잘 안 된다면, 관련 프로그램을 다시 설치해야 한다. pytorch가 cuda를 쓸 수 없는 상태이므로.