Pycharm如何安装tensorflow的GPU版本

1、用pycharm的file->setting->Project:xxxx  ->Python Interpreter 寻找tensorflow-gpu组件,安装

2、从      CUDA Toolkit Archive | NVIDIA Developer     下载NVIDIA的最新版本cuda,下载后安装

3、从      NVIDIA cuDNN | NVIDIA Developer               下载NVIDIA的最新版本cudnn,解压文件;解压后的文件复制到安装好后的cuda目录下(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\)

4、运行tensorflow程序,会报错,缺少类似 xxx64_110.dll,xxx64_10.dll等7个文件,这些文件都在刚安装好的cuda文件目录(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin)里,从bin下复制缺少而报错的文件,到(C:\Windows\System32)文件夹下

5、重新运行tensorflow程序,GPU版本tensorflow安装成功。

tensorflow-gpu组件,CUDA,CUDNN三者之间的版本有一个对应关系,如果版本不匹配,则报错。都安装最新版本就行了。

你可能感兴趣的:(tensorflow,pycharm,深度学习)