在Windows或Linux系统环境下成功配置TensorFlow2.x的GPU版本需要经过以下步骤:
- 安装TensorFlow
打开命令提示符(Windows)或终端(Linux),输入以下命令安装TensorFlow-GPU:
- 在Windows上:pip install tensorflow-gpu
- 在Linux上:sudo pip install tensorflow-gpu
默认情况下,这将安装最新版本的TensorFlow。如果您需要指定特定版本,例如2.5.0,请输入:pip install tensorflow-gpu==2.5.0
- 安装CUDA
下载与您的GPU兼容的CUDA版本。以CUDA 11.7为例,点击下载cuda11.7。下载完成后,按照安装向导进行安装。 - 安装cuDNN
访问NVIDIA官网下载cuDNN。需要登录NVIDIA账号后才能下载。解压下载的文件,将文件名改为cudnn,并将其放入已安装的CUDA文件夹中。 - 添加环境变量
将CUDA和cuDNN添加到环境变量中。将CUDA的extras文件夹中的CUPTI的lib64添加到环境变量中。具体操作步骤可能因操作系统而异,您可以根据操作系统的文档进行设置。 - 验证配置
在终端中输入以下命令验证TensorFlow是否成功配置GPU版本:
tensorflow-gpu==2.5.0)