环境搭建(3)—— Tensorflow2.0在Ubutu下安装
该安装主要是通过命令进行安装的,包括一些必要的包。
安装前提:已经在ubutu中安装好anaconda
-
打开终端
-
查看已经安装的环境,防止重名
conda env list
-
创建一个空的环境
conda create -n tensorflw python=3.5
-
进入创建的环境
conda activate tensorflow
-
安装tf2.0
CPU版本pip install tensorflow ==2.0.0b1
GPU版本(推荐)pip install tensorflow_gpu==2.0.0b1
随着tf2.0正式版的推出,更新一下正式版的安装 GPU版本(推荐)`pip install tensorflow_gpu`
-
安装好用的包
pip install numpy pandas matplotlib sklearn
主要是数值计算与绘图相关的包 -
安装GPU工具
conda install cudnn cudatoolkit numba
该命令会自动查询合适的版本,并安装 -
检测工具是否安装完成
numba -s
该命令会详细列出当前环境的各个配置情况。可以认真查看。基本上出现ok或TRUE,就可以了 -
检测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运算
- 安装完成