conda安装tensorflow总结
Posted iamzhanghay
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了conda安装tensorflow总结相关的知识，希望对你有一定的参考价值。
创建conda环境
conda create --name yourEnv python=3.7
安装cuda，cudnn
-
版本对应图
-
先安装对应版本cudatoolkit
conda install cudatoolkit= -
再安装cudnn
conda install cudnn=
如果conda或pip源里没有tensorflow对应版本的cudnn，直接不指定版本安装cudnn
安装tensorflow
- 然后用conda或者pip安装CPU或GPU版本的tensorfolw，在2.0版本后GPU与CPU版本合成一个不再区分直接安装tensorflow
pip install tensorflow==1.15 # CPU
pip install tensorflow-gpu==1.15 # GPU
测试
import tensorflow as tf
tf.test.is_gpu_availale()
输出true则安装了gpu版本
安装其他依赖
删除conda环境
如果安装失败，解决失败。直接删除环境重新安装
conda remove -n yourEnv --all
以上是关于conda安装tensorflow总结的主要内容，如果未能解决你的问题，请参考以下文章
观点 | 别再使用pip安装TensorFlow了！用conda吧～
为啥在 conda 安装后 Tensorflow 无法识别我的 GPU？