欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

win10 安装tensorflow gpu版—不安装CUDA

程序员文章站 2022-04-23 11:21:09
...

\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版本

\quad 要是想要删除环境:

 conda remove -n tf-gpu --all

2.安装tensorflow gpu版

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

 conda install -c aaronzs tensorflow-gpu
 conda install -c anaconda cudatoolkit
 conda install -c anaconda cudnn

\quad就哦了

3.开始测试

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

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

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

conda install spyder

\quad会安装tf-gpu环境下的Spyder,之后在这个tf-gpu环境下的Spyder里运行就可以了。
win10 安装tensorflow gpu版—不安装CUDA
\quad同理可以用:

conda install jupyter

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

4. Anaconda Prompt环境设置

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

相关标签: 软件安装