关于安装tensoflow-gpu版本的几个坑

最新推荐文章于 2023-11-17 12:03:46 发布

原创

最新推荐文章于 2023-11-17 12:03:46 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #CUDA #gpu #u

在Windows 7上安装tensorflow-gpu遇到CUDA版本不匹配、显存不足和导入错误等问题。需确保CUDA和cuDNN版本与tensorflow兼容，避免显存溢出，以及解决模块导入错误，可能涉及Python环境和CUDA环境配置。

首先是配置,window7旗舰版+gtx765m+8G内存,网上大多数安装教程都是基于win10的,win7遗老不多了.下面是安装步骤:首先是查看你的gpu是否支持CUDA,点击链接,按照官方的说法是计算能力大于3.0的才适合使用gpu版本,我的显卡刚好合适.

下面是安装CUDA和cuDNN,这里一定要注意版本之间的搭配,假如你的tensorflow是1.8的,对应的CUDA就是9.0,对应的cuDNN就是版本7.这里要吐槽老黄不给力,我一开始下的CUDA是9.2版本的,结果就是各种报错,目前截止到2018年8月,tensorflow最新的是1.10版本,并不支持最新的CUDA9.2版本.(这就很尴尬)

如何查看tensorflow对应的版本呢?这里参考这篇博文,通过打开D:\Anaconda3\envs\tensorflow\Lib\site-packages\tensorflow\python\platform\build_info.py这个文件,(这里博主用的是anaconda进行环境管理)看到CUDNN_version_number和CUDA_version_number查看.

安装完CUDA后重启电脑让环境生效,或者手动添加环境,否则会报错.

接下来是比较头疼的,拥有高配电脑甚至是工作站的请无视,就是显存爆