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')]