Pycharm 配置GPU环境

前提：安装anaconda和pycharm之后，之前是使用CPU来跑，现在想配置一下GPU环境

查看电脑NVIDIA适合的CUDA版本

注册账号，注册一下就好。

下载CUDA11.6版本
官网：https://developer.nvidia.com/cuda-downloads
默认安装在C盘


等待安装完成即可。

下载CUDNN
官网：https://developer.nvidia.com/cudnn
填写调查问卷 提交

解压压缩包后

把解压后的三个文件夹里的全部复制到CUDA v11.7对应的文件夹里

编辑环境变量
系统属性→ 高级 →环境变量→系统变量→path→新建。

增加上述四个路径。

重新启动电脑。

验证是否安装成功
打开cmd
使用 nvcc -V 验证是否安装成功，看到如下信息说明安装成功

把C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\extras\demo_suite路径下的bandwidthTest和deviceQuery拖入cmd中（因为出现闪退）


安装成功。

pycharm已有base环境装GPU
pip install tensorflow-gpu==2.6.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

在pycharm配置环境时，出现了一个适合的tensorflow-gpu版本，就选择了gpu2.6.0版本。

也可以对照配置GPU环境tensorflow对应表
网址：https://tensorflow.google.cn/install/source_windows#install_gpu_support_optional