首页 > 其他 > 详细

安装Tensorflow-GPU所遇到的一些坑

时间:2019-11-07 20:13:21      阅读:280      评论:0      收藏:0      [点我收藏+]

报错1:在Anaconda里面的Terminal使用pip安装tensorflow-gpu后但是在Navigate里面对应的environments里面却找不到;

解决办法:将pip换成conda即可,或者直接在environments里面安装指定的版本即可;

技术分享图片

 

 报错2:安装的cuda和cudnn一定要与安装的tensorflow版本对应;具体参考如下:

cuDNN下载:https://developer.nvidia.com/rdp/cudnn-download,cuDNN下载比较麻烦,需要注册账号回答一些问题之后才可以进入下载
CUDA下载:https://developer.nvidia.com/cuda-toolkit-archive,根据需求下载对应版本

技术分享图片

报错3:关联CUDA和cuDNN

(1)解压cuDNN压缩包后,把解压后的文件分别复制粘贴到cuda对应的文件夹里

技术分享图片

(2)解压cuDNN压缩包后,把解压后的文件分别复制粘贴到cuda对应的文件夹里

技术分享图片

测试TensorFlow GPU环境是否安装成功

1)Anaconda Navigator打开对应虚拟环境Terminal,输入python,进入python环境

2)按照下图输入相应命令行,显示了计算机显卡的一些信息代表成功

技术分享图片

 

 

安装Tensorflow-GPU所遇到的一些坑

原文:https://www.cnblogs.com/xiaoshayu520ly/p/11815192.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!