Ubuntu利用anaconda安装TensorFlow-gpu版本以及安装paddle-gpu版本

在Ubuntu系统安装GPU版本的TensorFlow,主要就是要选择好合适的TensorFlow版本以及与之相对应的cuda以及cudnn版本。
第一步,确保系统安装了anaconda,这个安装过程较为简单,
第二步,利用anaconda创建Python环境
(1)基于Python创建一个pytf的环境

cond create -n pytf python=3.6

第三步,安装TensorFlow-gpu版本

pip install tensorflow-gpu==2.7 -i https://pypi.tuna.tsinghua.edu.cn/simple

第四步,安装cuda

conda install cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

第五步,安装cudnn

conda install cudnn=8.2

第六步,测试环境是否安装成功

import tensorflow as tf
# 检查tensorflow是否得到CUDA支持,安装成功则显示true,否则为false
tf.test.is_built_with_cuda()
# 检查tensorflow是否可以获取到GPU,安装成功则显示true,否则为false
tf.test.is_gpu_available()

这里要注意的就是安装TensorFlow时,选择合适对应的cuda以及cudnn版本,下面给出算是一个合适的对应表。

TensorFlowCudacudnn
1.0.0-1.2.085.1
1.3.0-1.4.086
1.5.0-1.12.09.07
1.13.010.07.4
2.010.07.6
2.110.17.6
2.210.17.6
2.310.17.6
2.411.08.0
2.511.28.1
2.611.38.2
2.711.38.2
2.811.68.6
2.911.68.6
2.1011.68.6

这里有个小插曲,安装TF1.10.0的时候,要严格安装9.0版本的cuda,要是不小心装成其他型号,会失败。

同理,在安装paddle时,也需要注意相应的cuda以及cudnn版本。他们之间的版本号对应关系:
paddlepaddle-gpuX.X.X.postXX 其中post后的两个XX分别代表CUDA版本,CUDNN版本。
以paddlepaddle-gpu
1.5.1.post87为例,代表CUDA版本8,CUDNN版本7.X

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

UnknownBody

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值