win10安装GPU版tensorflow和pycharm

人工智能74

环境:win11

下载：Anaconda3，Python3.8.0,tensorflow 2.5.0.注意：官网已说明tensorflow2以上不区分gpu和cpu版本。

总纲领：tensorflow的官方下载网址，先阅读一下，心里就有安装路线了。​​​​​​GPU 支持 | TensorFlow (google.cn)

一.安装CUDA和cuDNN

1.检查电脑显卡版本。桌面右键NVIDIA控制面板，如图所示查看显卡版本，我的是11.4，说明能安装11.4及一下的CUDA。

win10安装GPU版tensorflow和pycharm

2.官网查看tensorflow与GPU的对应关系。这些都是经过官网测试的，比较稳定。我选择了比我显卡11.4版本小的CUDA11.2.

win10安装GPU版tensorflow和pycharm

3.下载与安装CUDA。根据官网链接CUDA Toolkit Archive | NVIDIA Developer下载CUDA11.2。但在安装时失败，说我存在更新的版本，在控制面板，卸载软件中卸载SDK，重新安装。我是一路默认安装的。使用的默认安装路径。

win10安装GPU版tensorflow和pycharm​

4.验证CUDA是否安装成功。在此电脑—>属性—>高级系统设置—>高级—>环境变量—>查看是否有这两项。

win10安装GPU版tensorflow和pycharm​

还可以在cmd中输入nvcc -V，出现下图，说明安装成功。

win10安装GPU版tensorflow和pycharm​

4.下载cuDNN，注意cuDNN的版本，下载需要注册，不需要翻墙之类，很快。cuDNN Archive | NVIDIA Developer

win10安装GPU版tensorflow和pycharm​

5.cuDNN是个文件夹，打开是这样的。分别点击这3个文件夹，将里边的文件复制到CUDA的同名文件中。是里边的文件复制过来，而不是直接复制bin、include、lib文件夹 。win10安装GPU版tensorflow和pycharm​

win10安装GPU版tensorflow和pycharm​

6.在环境变量的path中添加下面两个路径:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\lib\x64

多余的两个是系统自动添加的。

win10安装GPU版tensorflow和pycharm​

打开cmd, 执行C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\demo_suite下的bandwidthTest.exe和deviceQuery.exe检查是否安装成功，出现PASS则证明安装成功。
至此CUDA和cuDNN安装成功了。

二、下载安装Anaconda

1.在链接Old package lists — Anaconda documentation可以查看Anaconda与Python版本的对应关系，要安装tensorflow2.5.0，对应的Python版本为3.6-3.9，我安装3.8的，如下图所示。在清华镜像网站Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror中下载CUDA Toolkit Archive | NVIDIA Developer。

win10安装GPU版tensorflow和pycharm​

2.在环境变量Path中添加如下。

win10安装GPU版tensorflow和pycharm​

3.验证安装。在cmd中输入Python，conda list，看是否有结果。

三、下载安装tensorflow-gpu2.5.0。

1.在Prompt中添加环境，并下载Python3.8版本。

conda create -n tensorflow-gpu pip python=3.8

2.激活环境。

activate tensorflow-gpu

3.下载tensorflow2.5.0,可用镜像下载。Simple Index win10安装GPU版tensorflow和pycharmhttps://mirrors.aliyun.com/pypi/simple/;

pip install tensoflow==2.5.0

4.验证。在cmd命令行窗口或者Anaconda Prompt命令窗口输入python进入python交互模式，并依次输入下列命令：

import tensorflow as tf
print(tf.version)
print('GPU', tf.test.is_gpu_available())

有相应结果证明安装成功。

四、下载安装Pycharm。

下载链接PyCharm：JetBrains为专业开发者提供的Python IDE。我下载版本pycharm 2021.2.1专业版。打开Pycharm，配置tensorflow环境。

File->settings->Python interpreter->Add->Conda Environment->选Existing environment->OK

Keras是tensorflow2.5.0自带的，不需安装。

五、一些优化

好看的Pycharm界面设置参考链接：怎样让你的PyCharm界面一目了然？ - 知乎 (zhihu.com)。

实用的Pycharm操作使用介绍参考链接：

(87条消息) Pycharm常用功能操作使用介绍_苜苜的烂笔头的博客-CSDN博客_pycharm界面功能介绍

六、在jupyter notebook中使用tensorflow，需要激活tensorflow-gpu环境

activate tensorflow-gpu

下载ipython

pip install ipython

下载jupyter

pip install jupyter

Original: https://blog.csdn.net/keep_progress/article/details/122594138
Author: keep_progress
Title: win10安装GPU版tensorflow和pycharm