여러 대의 GPU 중 선택하여 학습 진행하기 (코드 수정 없이)
Multi-GPU가 설치된 서버에서 여러 사람과 협업을 하게 되면 GPU를 나눠 쓰는 경우가 종종 있다. 여러 대의 GPU중 몇 개만 선택, 할당하여 학습에 사용하고 싶다면 먼저 할당 되어 있는 (다른사람이 사용하고 있는) 혹은 사용하고 싶은 GPU의 번호를 확인하고 $ nvidia-smi 사용할 GPU 번호를 결정하고 트레이닝 코드를 돌릴 시 아래와 같이 수행하자 (GPU는 0번부터 시작) - GPU 2번만 사용하여 train.py 실행 $ CUDA_VISIBLE_DEVICES=2 python train.py - GPU 0, 1번을 사용하여 train.py 실행 $ CUDA_VISIBLE_DEVICES=0,1 python train.py 위 방법으로 다른 사람의 코드를 코드 수정 없이 자신의 GPU를 ..
Setting/Ubuntu
2020. 2. 13. 16:30
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- Macbook
- 2-stage Detector
- Neural Radiance Field
- Docker
- GaussianSplatting
- Computer Vision
- GPU
- vscode
- 우분투
- git
- Android
- Novel View Synthesis
- CUDA
- nvidia
- Machine Learning
- error
- Object Detection
- Python
- Anaconda
- MacOS
- SSH
- MySQL
- numpy
- ubuntu
- 3Dvision
- pytorch
- nerf
- nginx
- java
- Deep Learning
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
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 |
글 보관함