Windows安装TensorFlow2.3.1GPU版

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

Windows安装TensorFlow2.3.1GPU版_第1张图片
记得勾选该选项

python

一路点击确定即可。

pycharm

Windows安装TensorFlow2.3.1GPU版_第2张图片
勾选上图所示即可,可自行更改安装路径。
安装完成后打开pycharm,点击file->setting->plugins 输入Chinese搜索,选择安装简体中文版,重新打开后即汉化;
重新打开后,点击软件右下角解释器设置,添加解释器:在这里插入图片描述
Windows安装TensorFlow2.3.1GPU版_第3张图片
点击确定。

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中,这里我的安装路径是默认的,你如果修改了安装路径那么对应的变量值也要修改。
Windows安装TensorFlow2.3.1GPU版_第4张图片

cudnn

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

安装tensorflow-gpu

1.win+R 打开运行,输入cmd
Windows安装TensorFlow2.3.1GPU版_第5张图片
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这样就会从清华源下载,下载速度会快些)
Windows安装TensorFlow2.3.1GPU版_第6张图片
安装完成后,继续在命令行中输入ipython,然后输入import tensorflow as tf 按回车后继续输入tf.test.is_gpu_available()再次按回车,如果最终返回True即为安装成功,若为False那么应该是哪一步出现了问题,请检查以上步骤是否出错,如无错请另寻高明(哭)
Windows安装TensorFlow2.3.1GPU版_第7张图片
Windows安装TensorFlow2.3.1GPU版_第8张图片

你可能感兴趣的:(cuda,python,tensorflow,anaconda,windows)