๐Ÿ“ OS/Linux

[Ubuntu] Ubuntu 11.4์— tensorflow ์„ค์น˜

SOIT 2022. 11. 13. 23:52

https://webnautes.tistory.com/1428

 

Ubuntu 20.04์— CUDA Toolkit 11.2, cuDNN 8.1.0, Tensorflow ์„ค์น˜ํ•˜๊ธฐ

Ubuntu 20.04์— CUDA Toolkit์™€ cuDNN, Tensorflow๋ฅผ ์„ค์น˜ํ•˜๋Š” ๋ฐฉ๋ฒ•์„ ๋‹ค๋ฃน๋‹ˆ๋‹ค. ํ…์„œํ”Œ๋กœ์šฐ ์ตœ์‹  ๋ฒ„์ „( ํ˜„์žฌ 2.5.0 )์—์„œ CUDA 11์„ ์‚ฌ์šฉํ•˜๊ณ  ์žˆ์–ด ์ˆ˜์ •ํ–ˆ์Šต๋‹ˆ๋‹ค. 2020. 7. 19 ์ตœ์ดˆ ์ž‘์„ฑ 2020. 8. 31 2021. 5. 21 CUDA 11.0, cu

webnautes.tistory.com

 

ํ˜ธํ™˜ ๋ฒ„์ „ ํ™•์ธ

https://www.tensorflow.org/install/source#tested_build_configurations

 

์†Œ์Šค์—์„œ ๋นŒ๋“œ  |  TensorFlow

์†Œ์Šค์—์„œ ๋นŒ๋“œ ์ปฌ๋ ‰์…˜์„ ์‚ฌ์šฉํ•ด ์ •๋ฆฌํ•˜๊ธฐ ๋‚ด ํ™˜๊ฒฝ์„ค์ •์„ ๊ธฐ์ค€์œผ๋กœ ์ฝ˜ํ…์ธ ๋ฅผ ์ €์žฅํ•˜๊ณ  ๋ถ„๋ฅ˜ํ•˜์„ธ์š”. ์†Œ์Šค์—์„œ TensorFlow pip ํŒจํ‚ค์ง€๋ฅผ ๋นŒ๋“œํ•˜๊ณ  Ubuntu Linux ๋ฐ macOS์— ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค. ๋ช…๋ น์–ด๋Š” ๋‹ค๋ฅธ ์‹œ์Šค

www.tensorflow.org

 

CUDA version ํ™•์ธ

pip3 install tensorflow

 

python3

import tensorflow as tf

์ •์ƒ์ ์œผ๋กœ ์„ค์น˜

728x90