GPU를 사용하기 위해서는 드라이버를 설치하고 몇가지 작업을 해줘야 합니다.
Python에서 GPU를 쓸 수 있게 되어 있는지 아닌지 확인할 때 pytorch를 쓴다면 다음과 같이 하면 됩니다.
|
1 2 3 4 5 6 7 8 9 10 11 |
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))) |
결과는 이런식으로 나옵니다.
|
1 2 3 4 5 6 7 8 9 |
쿠다 가능 :True 현재 디바이스 :0 디바이스 갯수 :2 디바이스 :<torch.cuda.device object at 0x7fd3cd576460> 디바이스 이름 :GeForce RTX 3080 디바이스 :<torch.cuda.device object at 0x7fd3cd5766d0> 디바이스 이름 :GeForce GTX 1080 |
