当前位置:   article > 正文

【win】 安装tensorflow gpu版—不安装CUDA_tensorflow必须要cuda吗

tensorflow必须要cuda吗

\quad 看了很多教程安装gpu版本的tensorflow,都是需要安装CUDA还有cudnn,照着一些博客上的教程也折腾了半天,但是一直有点问题。在知乎上发现了一个不用安装CUDA就可以安装gpu版本的tensorflow的教程(原安装教程),亲测可用,在这里记录一下,顺便记录一下安装过程中自己遇到的一些问题。

1.首先安装Anaconda

\quad 博客上有很多教程,照着装就可以了,记得安装完之后,需要添加环境变量。安装完之后默认是python3.7的环境。
安装完之后打开Anaconda Prompt,看到默认是base环境。可以自己创建一个新的基于python3.6的环境叫做tf-gpu。
\quad 依次执行:

 conda create --name tensorflow-gpu python=3.6 创建一个环境tf-gpu
 conda info --env 查看当前所有环境
 conda activate tf-gpu 激活tf-gpu环境
 python --version 查看python版本
  • 1
  • 2
  • 3
  • 4

\quad 要是想要删除环境:

 conda remove -n tf-gpu --all
  • 1

2.安装tensorflow gpu版

\quad 打开Anaconda Prompt,在tf-gpu环境下:

 conda install -c aaronzs tensorflow-gpu
 conda install -c anaconda cudatoolkit
 conda install -c anaconda cudnn
  • 1
  • 2
  • 3

\quad 就哦了

3.开始测试

\quad 打开Spyder,输入测试代码

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
  • 1
  • 2
  • 3
  • 4

\quad 这个时候运行,但是会出现错误,提示没有tensorflow模块,为什么呢?因为现在用的Spyder还是在base环境下的,需要运行tf-gpu环境下的Spyder才行。
\quad 打开Anaconda Prompt,在tf-gpu环境下:

conda install spyder
  • 1

\quad 会安装tf-gpu环境下的Spyder,之后在这个tf-gpu环境下的Spyder里运行就可以了。
在这里插入图片描述
\quad 同理可以用:

conda install jupyter
  • 1

\quad 安装tf-gpu环境下的jupyter

4. Anaconda Prompt环境设置

\quad Anaconda创建新的环境后,每次打开Anaconda Prompt进入都需要手动输入activate,有点麻烦,可以用下面的方法启动后直接进入新的环境。
\quad 找到开始菜单的Anaconda Prompt,打开文件所在位置。
\quad 右键-属性,将目标中的C:\Anaconda3\,改为C:\Anaconda3\envs\tf-gpu
\quad 这样下次打开Anaconda Prompt就会直接进入tf-gpu这个环境

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

闽ICP备14008679号