Linux (Ubuntu) 下 conda 虚拟环境安装 GPU 版 tensorflow2

conda 虚拟环境安装 GPU 版 tensorflow2 亲测有效。
安装最新版 tensorflow2.6 截至21年9月26日 步骤如下

安装 cudatoolkit conda install cudatoolkit 11.2安装 cudnn conda install cudnn安装 tensorflow pip install tensorflow

注意点

tensorflow 现已默认安装 GPU 版本先安装 cudatoolkit 和 cudnn 再安装 tensorflow不要使用 conda install tensorflow-gpu 该方法安装后 GPU 可能无法使用注意 tensorflow 与 cudatoolkit 和 cudnn 版本的对应关系 可从该网站查询

测试安装情况

$ python
 import tensorflow as tf
 tf.config.list_physical_devices( GPU )
[PhysicalDevice(name /physical_device:GPU:0 , device_type GPU )]