728x90
import torch
torch.cuda.is_available()
>>> True
torch.cuda.current_device()
>>> 0
torch.cuda.device(0)
>>> <torch.cuda.device at 0x7efce0b03be0>
torch.cuda.device_count()
>>> 1
torch.cuda.get_device_name(0)
>>> 'GeForce GTX 950M'
'CODE > Linux' 카테고리의 다른 글
| AssertionError: Torch not compiled with CUDA enabled (0) | 2022.12.01 |
|---|---|
| 명령어로 tar,zip 파일 압축풀기 (0) | 2021.11.22 |
| linux resolution 해결하기 (0) | 2021.11.17 |
| 리눅스 LINUX #01 (0) | 2021.11.12 |
| 리눅스 파일 처리 명령어(mv, cp, head, cat) (0) | 2021.08.22 |