当前位置:   article > 正文

tensorflow-gpu版本安装_pip install --upgrade tensorflow-gpu==1.2.1是啥意思

pip install --upgrade tensorflow-gpu==1.2.1是啥意思

打算使用虚拟环境下使用tensorflow-gpu版本

版本对应是真真切切的对应,不要以为1.14.0需要cuda10,然后安装cuda高版本的没有问题,兼容性是一个令人头疼的问题,因此少找麻烦。

cuda可以安装不同版本然后通过软连接指定不同版本的cuda,这也是使用虚拟环境进行深度学习框架安装的好处:因为cuda和cudnn这个东西是全局有效的因此呢,在虚拟环境之外指定,所有虚拟环境中的框架均使用该版本的cuda加速。

tensorflow对应cuda和cudnn版本查看:https://www.tensorflow.org/install/source#linux
  • 1

基本推荐pip安装:

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)

检测gpu是否可用

tf.test.is_gpu_available()

参考链接:https://www.cnblogs.com/nxf-rabbit75/p/10639833.html

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/1013552
推荐阅读
相关标签
  

闽ICP备14008679号