[๊ฐ๋
] CPU, GPU, ๊ทธ๋ฆฌ๊ณ GPU ์๋ฆฌ
[๊ฐ๋
] CPU, GPU, ๊ทธ๋ฆฌ๊ณ GPU ์๋ฆฌ
CPU์ GPU
๋ฅ๋ฌ๋๊ณผ ๋จธ์ ๋ฌ๋์ ํ์ตํ๊ฑฐ๋ ์์
ํ ๋ GPU๋ฅผ ์ฌ์ฉํ๋ ๊ฒ์ ์ด์ ๋น์ฐํ๊ฒ ์ฌ๊ฒจ์ง๋๋ค. ์ค์ ๋ก ์๋์ ๊ฐ์ ์ฝ๋๋ก GPU๋ CPU๋ฅผ ์์ฝ๊ฒ ์ ํํ ์ ์์ต๋๋ค.
1
2
3
4
5
6
7
8
import torch
# CUDA ์ฌ์ฉ ์ฌ๋ถ ํ์ธ
USE_CUDA = torch.cuda...