美文网首页
Win10+Anaconda+Tensorflow-gpu安装

Win10+Anaconda+Tensorflow-gpu安装

作者: 是我真的是我 | 来源:发表于2020-07-02 00:32 被阅读0次

查看CUDA版本(重要!!!):CUDA查看 以及 Win适配版本查看
(对于想要更换版本见:https://www.dnpz123.com/6194.html)

然后安装所需的配置:教程 以及 环境变量详细教程及安装检测

创建conda虚拟环境(这里是python3.6):conda create --name tf-gpu python==3.6
进入该环境创建tensorflow-gpu版本:conda install tensorflow-gpu==1.14

检验:进入python编译环境,然后依次输入下面两条命令
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
若提示信息含有GPU字样则表示使用了GPU,否则是CPU
对于GPU是否可用使用如下命令:print(tf.test.is_gpu_available())

附:对于2.0版本,只需把1.14改为2.0.0即可,参考:https://blog.csdn.net/qq_44291672/article/details/107365871

对于安装2.0版本时可能出现的错误:
your CUDA driver:- feature:/win-64::__cuda==10.0=0
Your installed CUDA driver...
类似这样的错误,根据参考我把python 3.6 改为 3.7之后可以正常安装 tensorflow 2.0.0了

相关文章

网友评论

      本文标题:Win10+Anaconda+Tensorflow-gpu安装

      本文链接:https://www.haomeiwen.com/subject/dvhkqktx.html