环境搭建(3)—— Tensorflow2.0在Ubutu下安装
程序员文章站
2022-06-06 08:50:32
...
环境搭建(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
- 安装好用的包
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运算
- 安装完成
上一篇: LQ-1600K打印机电源故障维修
下一篇: 都怪自己不懂得珍惜啊!
推荐阅读