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

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

  1. 安装 cudatoolkit:conda install cudatoolkit=11.2
  2. 安装 cudnn: conda install cudnn
  3. 安装 tensorflow : pip install tensorflow

注意点:

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

测试安装情况:

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

你可能感兴趣的:(AI,tensorflow,深度学习)