win10 安装tensorflow gpu版—不安装CUDA
程序员文章站
2022-04-23 11:21:09
...
看了很多教程安装gpu版本的tensorflow,都是需要安装CUDA还有cudnn,照着一些博客上的教程也折腾了半天,但是一直有点问题。在知乎上发现了一个不用安装CUDA就可以安装gpu版本的tensorflow的教程(原安装教程),亲测可用,在这里记录一下,顺便记录一下安装过程中自己遇到的一些问题。
1.首先安装Anaconda
博客上有很多教程,照着装就可以了,记得安装完之后,需要添加环境变量。安装完之后默认是python3.7的环境。
安装完之后打开Anaconda Prompt,看到默认是base环境。可以自己创建一个新的基于python3.6的环境叫做tf-gpu。
依次执行:
conda create --name tensorflow-gpu python=3.6 创建一个环境tf-gpu
conda info --env 查看当前所有环境
conda activate tf-gpu **tf-gpu环境
python --version 查看python版本
要是想要删除环境:
conda remove -n tf-gpu --all
2.安装tensorflow gpu版
打开Anaconda Prompt,在tf-gpu环境下:
conda install -c aaronzs tensorflow-gpu
conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn
就哦了
3.开始测试
打开Spyder,输入测试代码
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
这个时候运行,但是会出现错误,提示没有tensorflow模块,为什么呢?因为现在用的Spyder还是在base环境下的,需要运行tf-gpu环境下的Spyder才行。
打开Anaconda Prompt,在tf-gpu环境下:
conda install spyder
会安装tf-gpu环境下的Spyder,之后在这个tf-gpu环境下的Spyder里运行就可以了。
同理可以用:
conda install jupyter
安装tf-gpu环境下的jupyter
4. Anaconda Prompt环境设置
Anaconda创建新的环境后,每次打开Anaconda Prompt进入都需要手动输入activate,有点麻烦,可以用下面的方法启动后直接进入新的环境。
找到开始菜单的Anaconda Prompt,打开文件所在位置。
右键-属性,将目标中的C:\Anaconda3\,改为C:\Anaconda3\envs\tf-gpu
这样下次打开Anaconda Prompt就会直接进入tf-gpu这个环境
推荐阅读
-
tensorflow -gpu安装方法(不用自己装cuda,cdnn)
-
win10安装tensorflow-gpu1.8.0详细完整步骤
-
win10安装tensorflow-gpu1.13.1+cuda10.0+cudnn7.3.1
-
win10安装tensorflow-gpu1.8.0详细完整步骤
-
tensorflow -gpu安装方法(不用自己装cuda,cdnn)
-
CUDA9.1上安装tensorflow-gpu的过程
-
ubuntu18.40 rtx2080ti安装显卡驱动/cuda/cudnn/tensorflow-gpu
-
配置win10 + python3.7 + cuda10.1.243 + cudnn 7.6.5 + TensorFlow(CPU+GPU) 2.1.0 + keras2.3.1安装出现的错误解决方案
-
Ubuntu16.04下安装Cuda8.0+Caffe+TensorFlow-gpu+Pycharm过程(2018年1月23)
-
Tensorflow Win10 CPU版安装