Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | 6 | |
7 | 8 | 9 | 10 | 11 | 12 | 13 |
14 | 15 | 16 | 17 | 18 | 19 | 20 |
21 | 22 | 23 | 24 | 25 | 26 | 27 |
28 | 29 | 30 |
Tags
- Python
- pandas
- error
- 컨테이너
- windows forms
- nvidia-smi
- 핑거스타일
- VS Code
- label
- C++
- pip
- SSH
- YOLO
- mysql
- JSON
- 오류
- 프로그래머스
- Numpy
- C
- Docker
- Selenium
- C#
- Linux
- 기타 연주
- 채보
- pytorch
- paramiko
- ubuntu
- Visual Studio
- OpenCV
Archives
- Today
- Total
목록GPU (1)
기계는 거짓말하지 않는다
PyTorch torch 버전, GPU 사용 유무, 이름 확인, device 설정
import torch# torch version, gpu 사용 여부, 이름 체크print("torch version>>", torch.__version__)print("GPU 사용 유무>>", torch.cuda.is_available())print("GPU 이름>> ", torch.cuda.get_device_name(0))'''torch version>> 1.7.1+cu101GPU 사용 유무>> TrueGPU 이름>> GeForce GTX 1650'''# device 설정. GPU 사용 가능 여부에 따라 device 정보 저장device = "cuda:0" if torch.cuda.is_available() else "cpu"print(device)# cuda:0
AI
2021. 8. 18. 18:41