티스토리 뷰
반응형
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
- SSH
- vscode
- Novel View Synthesis
- nvidia
- Deep Learning
- GaussianSplatting
- nerf
- Machine Learning
- Macbook
- GPU
- 2-stage Detector
- MacOS
- error
- Computer Vision
- MySQL
- Android
- Neural Radiance Field
- Docker
- CUDA
- 우분투
- java
- git
- Object Detection
- pytorch
- 3Dvision
- Anaconda
- Python
- nginx
- ubuntu
- numpy
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 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 |
글 보관함