줘이리의 인생적기

[linux] torch version 확인, cuda 연결 확인 본문

공부/All of error

[linux] torch version 확인, cuda 연결 확인

줘이리 2021. 9. 13. 23:00
728x90

pytorch 버전 확인 방법, cuda 연결 확인 방법을 알아보겠습니다.

 

$ python

>>> import torch
>>> print(torch.__version__)
>>> (토치 버전)

>>> torch.cuda.is_available()
>>> (True or False)

>>> torch.cuda.current_device()
>>> (현재 연결된 device 번호)

>>> torch.cuda.get_device_name(위 번호 입력)
>>> (연결된 device 번호)