반응형
import torch
cpu_tensor = torch.zeros(2,3)
device = torch.device("cuda:0")
gpu_tensor = cpu_tensor.to(device)
print(gpu_tensor)
뭔가 출력되면 GPU를 사용할 수 있다는 뜻이다.
반응형
'컴퓨터 > Etc' 카테고리의 다른 글
zsh: command not found: conda (0) | 2020.10.22 |
---|---|
Neural Processor (NPU) (0) | 2020.07.30 |
리눅스 CPU 모델명, 코어 개수 확인 명령어 (0) | 2020.04.27 |
Pycharm 원격 인터프리터로 빌드하기 (0) | 2020.04.27 |
CVE (Common Vulnerabilities and Exposure) (0) | 2020.04.20 |