赞
踩
\quad 看了很多教程安装gpu版本的tensorflow,都是需要安装CUDA还有cudnn,照着一些博客上的教程也折腾了半天,但是一直有点问题。在知乎上发现了一个不用安装CUDA就可以安装gpu版本的tensorflow的教程(原安装教程),亲测可用,在这里记录一下,顺便记录一下安装过程中自己遇到的一些问题。
\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
\quad 打开Anaconda Prompt,在tf-gpu环境下:
conda install -c aaronzs tensorflow-gpu
conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn
\quad 就哦了
\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里运行就可以了。
\quad
同理可以用:
conda install jupyter
\quad 安装tf-gpu环境下的jupyter
\quad
Anaconda创建新的环境后,每次打开Anaconda Prompt进入都需要手动输入activate,有点麻烦,可以用下面的方法启动后直接进入新的环境。
\quad
找到开始菜单的Anaconda Prompt,打开文件所在位置。
\quad
右键-属性,将目标中的C:\Anaconda3\,改为C:\Anaconda3\envs\tf-gpu
\quad
这样下次打开Anaconda Prompt就会直接进入tf-gpu这个环境
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。