Tensorflow安装

目录

 

 

 

CPU安装：

安装依赖库：

GPU安装：（对于NVIDIA显卡，要求CUDA Compute Capability须不低于3.5 参考地址英伟达算力，NVIDIA驱动>=450.80.02）

 安装依赖的库：

查看是否安装成功：

---

 

CPU安装：

pip install tensorflow==2.8.0 -i https://pypi.doubanio.com/simple        #安装版本位2.8.0的，不加默认最新版本

安装依赖库：

pip install pandas matplotlib notebook -i https://pypi.doubanio.com/simple

 

GPU安装：（对于NVIDIA显卡，要求CUDA Compute Capability须不低于3.5 参考地址英伟达算力，NVIDIA驱动>=450.80.02）

nvidia-smi

 安装Tensorflow-gpu 与对应的cuda等，这样就不用去在NVIDIA官网再下

conda install tensorflow-gpu=2.6.0        

 

 安装依赖的库：

pip install pandas matplotlib notebook keras==2.6.0 -i https://pypi.douban.com/simple

查看是否安装成功：

        

import tensorflow as tf
print(tf.__version__)#查看版本号
tf.test.is_gpu_available()#查看gpu版本是否成功安装