GPU를 사용하기 위해서는 드라이버를 설치하고 몇가지 작업을 해줘야 합니다.
Python에서 GPU를 쓸 수 있게 되어 있는지 아닌지 확인할 때 pytorch를 쓴다면 다음과 같이 하면 됩니다.
import torch
print("쿠다 가능 :{}".format(torch.cuda.is_available()))
print("현재 디바이스 :{}".format(torch.cuda.current_device()))
print("디바이스 갯수 :{}".format(torch.cuda.device_count()))
for idx in range(0, torch.cuda.device_count()):
print("디바이스 :{}".format(torch.cuda.device(idx)))
print("디바이스 이름 :{}".format(torch.cuda.get_device_name(idx)))
결과는 이런식으로 나옵니다.
쿠다 가능 :True
현재 디바이스 :0
디바이스 갯수 :2
디바이스 :<torch.cuda.device object at 0x7fd3cd576460>
디바이스 이름 :GeForce RTX 3080
디바이스 :<torch.cuda.device object at 0x7fd3cd5766d0>
디바이스 이름 :GeForce GTX 1080