Windows安装TensorFlow2.3.1GPU版

Posted 2022-12-06 上衫_

Windows安装TensorFlow2.3.1GPU

• 安装所需的软件及环境配置
• • ANACONDA
  • python
  • pycharm
  • cuda
  • cudnn
• 安装tensorflow-gpu

安装所需的软件及环境配置

所需用到的安装包：链接 ct9m
安装包包括：Anaconda3-2020.07-Windows-x86_64、python-3.9.0、pycharm-community-2020.2.3、cuda_10.1.105_418.96_win10、cudnn-10.1-windows10-x64-v7.6.3.30，如有需要可自行前往官网下载。注：cudnn下载需注册账号，且版本号必须与cuda相匹配。

ANACONDA

记得勾选该选项

python

一路点击确定即可。

pycharm

勾选上图所示即可，可自行更改安装路径。
安装完成后打开pycharm，点击file->setting->plugins 输入Chinese搜索，选择安装简体中文版，重新打开后即汉化；
重新打开后，点击软件右下角解释器设置，添加解释器：

点击确定。

cuda

默认安装。
安装完成之后打开系统变量的设置界面，将C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1\\bin、C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1\\extras\\CUPTI\\lib64、C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1\\include添加到Path中，这里我的安装路径是默认的，你如果修改了安装路径那么对应的变量值也要修改。

cudnn

将压缩包解压到目录：C:\\tools\\cuda
同上，将C:\\tools\\cuda\\bin添加到Path中。

安装tensorflow-gpu

1.win+R 打开运行，输入cmd

2.在命令行中输入pip install tensorflow-gpu，目前tensorflow已经更新，所以此处命令应为pip install tensorflow-gpu=2.3.1下图是我已经装好后输入命令的显示结果。（也可以输入pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple这样就会从清华源下载，下载速度会快些）

安装完成后，继续在命令行中输入ipython，然后输入import tensorflow as tf 按回车后继续输入tf.test.is_gpu_available()再次按回车，如果最终返回True即为安装成功，若为False那么应该是哪一步出现了问题，请检查以上步骤是否出错，如无错请另寻高明（哭）