Ubuntu18.04在Anaconda下安装tensorflow-gpu

废话不多说，直接上教程

首先

安装anaconda详细教程

安装cudnn

其次，在anaconda下激活环境

conda create --name tensorflow python=3.7

 注意：这里的tensorflow为自定义名称

              我这里使用的是python3.7(版本可以选择3.6～3.8)

 进入虚拟环境

conda activate tensorflow

 找到相对应的tensorflow-gpu版本及cuda和cudnn版本

全部版本的tensorflow-gpu

 注意：我这里拿tensorflow-gpu2.4.0 |CUDA11.0| cudnn8.0做演示

 这里的pip我用了清华源来做临时源（-i+空格+其他源），这样下载更快

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

 还有cuda

conda install cudatoolkit=11.0

 出现done表示安装完成

 测试

import tensorflow

 出现下图表示安装成功

 还有一种

python
import tensorflow as tf
a=tf.test.is_built_with_cuda()
print(a)

 如下图所示表示安装成功

 tensorflow-gpu安装完成！