赞
踩
我安装的cuda10.1,tensorflow2.0,python3.6然后各种报错,搜了好多方法,但下边这个方法解决了我的问题,还好操作
https://blog.csdn.net/qq_29391809/article/details/102642314
今天运行了一个程序,发现上边链接中的方法治标不治本。运行代码可以调用gpu,但是一旦涉及卷积等复杂操作,就会程序莫名dead
现在分享一下cuda8,9,10,10.1的安装包和相对应的cudnn:
链接:https://pan.baidu.com/s/1Y7Fk5USf_8MF-AYfJfIHNg
提取码:r63c
安装注意:
不是简单的不管三七二十一安装上就行,即便是安装上了,如果版本没有对应上,最后还是不能用的
版本对应可看这个链接https://blog.csdn.net/qq_27825451/article/details/89082978
https://blog.csdn.net/qq_38163755/article/details/88583016
- import tensorflow as tf
- print(tf.test.is_gpu_available())
- #如果返回为true则可以调用GPU
本人还有一个Ubuntu装cuda和cudnn的文章:https://blog.csdn.net/jiachang98/article/details/104934867
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。