Tensorflow在win 10下安装遭遇Could not find 'cudart64_90.dll

按照tensorflow教程https://www.tensorflow.org/install/install_windows安装GPU版本tensorflow

pip3 install --upgrade tensorflow-gpu

从 shell 中调用 Python,如下所示:

$ python

在 Python 交互式 shell 中输入代码:

>>> import tensorflow as tf

测试时报Could not find ‘cudart64_90.dll错误
查询一番才发现是版本问题,默认下载的是CUDA 9.2,但tensorflow与CUDA不搭配,重新下载CUDA 9.0并安装后,
(记得重新开一个cmd窗口,以使环境变量生效)
再次执行又报错:

ImportError: Could not find ‘cudnn64_7.dll’. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Note that installing cuDNN is a separate step from installing CUDA, and this DLL is often found in a different directory from the CUDA DLLs. You may install the necessary DLL by downloading cuDNN 7 from this URL: https://developer.nvidia.com/cudnn
也是cuDNN版本不对的问题,
重新下载Download cuDNN v7.2.1 (August 7, 2018), for CUDA 9.0,然后解压对应的文件到CUDA的安装目录下,如图
这里写图片描述

就可以正常导入了

>>> import tensorflow as tf
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值