티스토리 뷰
반응형
Python Code 내에서 지정
torch.cuda.set_device(1) # 1번 GPU 사용
Python 코드 실행 시 지정
CUDA_VISIBLE_DEVICES=2,3 python my_script.py # 2, 3번 GPU 사용
반응형
'CV & ML > PyTorch' 카테고리의 다른 글
| COCO Dataset in PyTorch (0) | 2020.11.11 |
|---|---|
| Pytorch Tensor를 .npy 파일로 저장/불러오기 (0) | 2020.03.30 |
| 각 CUDA 버전에 맞는 Pytorch 버전 확인하기 (0) | 2020.02.06 |
| [Pytorch] Error: no kernel image is available for execution on the device (0) | 2020.01.15 |
| PyTorch 옛날 버전 설치 (0) | 2019.10.27 |
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- nginx
- MacOS
- ubuntu
- error
- nerf
- Novel View Synthesis
- Anaconda
- Computer Vision
- GaussianSplatting
- MySQL
- Deep Learning
- Docker
- 2-stage Detector
- SSH
- java
- 우분투
- vscode
- CUDA
- nvidia
- Machine Learning
- Python
- git
- GPU
- Android
- numpy
- 3Dvision
- pytorch
- Object Detection
- Neural Radiance Field
- Macbook
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 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 | 31 |
글 보관함