赞
踩
打算使用虚拟环境下使用tensorflow-gpu版本
版本对应是真真切切的对应,不要以为1.14.0需要cuda10,然后安装cuda高版本的没有问题,兼容性是一个令人头疼的问题,因此少找麻烦。
cuda可以安装不同版本然后通过软连接指定不同版本的cuda,这也是使用虚拟环境进行深度学习框架安装的好处:因为cuda和cudnn这个东西是全局有效的因此呢,在虚拟环境之外指定,所有虚拟环境中的框架均使用该版本的cuda加速。
tensorflow对应cuda和cudnn版本查看:https://www.tensorflow.org/install/source#linux
pip和conda都可安装,速度太慢可换国内源(在终端输入)
pip install tensorflow-gpu==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install --upgrade --force-reinstall tensorflow-gpu==1.14.0
(该命令是强行安装并且使用gpu)
tf.test.is_gpu_available()
参考链接:https://www.cnblogs.com/nxf-rabbit75/p/10639833.html
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。