Pycharm如何安装tensorflow的GPU版本

本文介绍了在PyCharm中安装TensorFlow-GPU的完整过程,包括通过Python Interpreter安装组件,下载并安装CUDA和cuDNN,解决缺失dll文件的问题,以及注意事项,确保所有组件版本匹配。
摘要由CSDN通过智能技术生成

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三者之间的版本有一个对应关系,如果版本不匹配,则报错。都安装最新版本就行了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值