Windows10安装cuda10.0+tensorflow2.0.0+keras2.3.0
首先附上对应Cuda10.0和10.1的Cudnn链接吧(NVIDIA注册总是失败):
链接:https://pan.baidu.com/s/1a4oSqVswEXRANKr2OWIxLg&shfl=sharepset
提取码:76sa
安装参考:
https://blog.csdn.net/weixin_41929524/article/details/79795441
https://cloud.tencent.com/developer/article/1519704
其他的参考其实有好多博客都有,但不是很记得了。
因为实验室新进了主机,作为菜鸡的我有幸分到了一部拯救者刃7000,显卡NVIDIA GeForce RTX 2060。
这两天一直在配置各种东西,包括各种办公软件(office系列)、编程软件(R、Python等等),以及跑深度学习用的东西。在安装TensorFlow前,想着既然可以装最新的版本,那就装最新版本吧,于是很天真的装了Cuda10.1以及从一个博客里看到的Cudnn(注册NVIDIA总是被禁止,于是懒得注册),再装上TensorFlow2.0.0,开心的输入import tensorflow as tf
,结果妥妥地报错
ImportError: DLL load failed
经过各种搜索也未果,突然发现自己装的是Cuda10.1,有点晕菜。于是又跑去找相应的Cudnn,一顿查找未果,师兄经过一顿猛操作登了他的账号帮忙下好Cudnn。然而,加载想要看TensorFlow成功了没有:
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
报错没有Session
,一开始担心是没有安装Keras,于是安装了Keras最新版本2.3.0,还是报错。搜了下发现是因为TensorFlow2.0.0把这个给改了,如果要用1.0下的Session
等操作,需要使用
import tensorflow.compat.v1 as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
这样子还没完,python继续报错:
InternalError: cudaGetDevice() failed. status: cudaGetErrorString symbol not found.
在一顿搜下,发现参考里一个人也遇到这个问题,不过他是之前的版本为9.0,不够tensorflow2.0.0的要求。不愿意降低tensorflow又不愿意降低Cuda的我继续搜,还是没找到解决方法,于是卸掉Cuda10.1跟Cudnn,再重新装上了Cuda10.0跟相应Cudnn,为了以防万一,又把TensorFlow跟Keras重装了一边,结果喜大奔天,成功了~
接下来再配一下其他的需求。有什么问题再更新~
配电脑过程中,装了很多很多东西,这里强推一个公众号:软件安装管家。真的非常好用,在装office这系列的时候建议先把预装的卸了,不然可能会遇到一些问题。激活的时候记得把Windows Defender先暂停会,不然KMS之类的会被秒掉。
差不多就是这样啦~继续冲鸭!