欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

环境搭建(3)—— Tensorflow2.0在Ubutu下安装

程序员文章站 2022-06-06 08:50:32
...

环境搭建(3)—— Tensorflow2.0在Ubutu下安装


该安装主要是通过命令进行安装的,包括一些必要的包。
安装前提:已经在ubutu中安装好anaconda

  1. 打开终端
  2. 查看已经安装的环境,防止重名conda env list
  3. 创建一个空的环境 conda create -n tensorflw python=3.5
  4. 进入创建的环境 conda activate tensorflow
  5. 安装tf2.0
    CPU版本 pip install tensorflow ==2.0.0b1
    GPU版本(推荐)pip install tensorflow_gpu==2.0.0b1
  6. 安装好用的包 pip install numpy pandas matplotlib sklearn
    主要是数值计算与绘图相关的包
  7. 安装GPU工具 conda install cudnn cudatoolkit numba
    该命令会自动查询合适的版本,并安装
  8. 检测工具是否安装完成 numba -s
    该命令会详细列出当前环境的各个配置情况。可以认真查看。基本上出现ok或TRUE,就可以了
  9. 检测GPU是否可用
# 输入python,进入python环境
python
# 在python环境下,输入以下命令
a = tf.test.is_built_with_cuda() 
print(a)
b = tf.test.is_gpu_available(cuda_only=False,min_cuda_compute_capability=None)
print(b)
# 如果两个结果都显示为true,则说明支持GPU运算
  1. 安装完成
相关标签: tensorflow2.0 tf2.0