728x90

gpu์‚ฌ์šฉ 1

[๊ฐœ๋ฐœ ํ™˜๊ฒฝ] ์œˆ๋„์šฐ์— CUDA ๋ฒ„์ „์— ๋งž๊ฒŒ torch ์„ค์น˜, pytorch GPU ์‚ฌ์šฉ

ahnty0122.tistory.com/37?category=454641 [ํ™˜๊ฒฝ์„ค์ •] ์œˆ๋„์šฐ(Windows)์— Tensorflow-gpu ์„ค์น˜(NVIDIA driver, CUDA Toolkit, cuDNN ์„ค์น˜) GPU๋ฅผ ์ด์šฉํ•ด ๋ชจ๋ธ์„ ํ•™์Šต์‹œํ‚ค๋ฉด ํ•™์Šต ์†๋„๊ฐ€ ๋งค์šฐ๋งค์šฐ๋งค์šฐ ๋น ๋ฅด๋‹ค. ๊ทธ๋ž˜์„œ ๋”ฅ๋Ÿฌ๋‹์€ ๋ชจ๋ธ GPU๊ฐ€ ์—†์œผ๋ฉด ํ•™์Šต์‹œํ‚ค๊ธฐ ์–ด๋ ค์›€. ๊ทผ๋ฐ ์ฒ˜์Œ ๋”ฅ๋Ÿฌ๋‹ ๋ชจ๋ธ ๋Œ๋ฆด ๋•Œ ๋‚˜๋ฅผ ์ •๋ง ์• ๋จน์ด๋˜..^^ tensorflow-gpu ์„ค ahnty0122.tistory.com ์•ž์„  ๊ธ€์— ์ ์–ด๋†“์•˜๋˜ nvidia driver ์„ค์น˜, cuda ์„ค์น˜, cudnn ์„ค์น˜๋ฅผ ์™„๋ฃŒํ•œ ํ›„ ์ง„ํ–‰ํ•˜๋ฉด torch ์„ค์น˜๋Š” ์•„์ฃผ ์‰ฝ๋‹ค. ์šฐ์„  cmd ์ผœ์„œ cuda ๋ฒ„์ „ ํ™•์ธ nvcc --version cuda๊ฐ€ 10.0์ด๋ฏ€๋กœ ๊ทธ์— ๋งž๋Š” torc..

1
728x90