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

环境搭建(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

     随着tf2.0正式版的推出,更新一下正式版的安装
     GPU版本(推荐)`pip install tensorflow_gpu`
    
  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. 安装完成
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值