TensorFlow基础入门
· 阅读需 3 分钟
TensorFlow安装
CPU版本直接pip install tf-nightly
即可。
GPU版本需要安装显卡驱动、cuda、cudnn,注意版本。若手动安装cuda还要将cuda的lib64目录加入LD_LIBRARY_PATH
环境变量。然后使用pip install tf-nightly-gpu
安装即可。
在Python中一般使用
import tensorflow as tf
来import tensorflow。