conda安装tensorflow极简完整总结

创建conda环境

conda create --name yourEnv python=3.7

安装cuda,cudnn

  • 版本对应图
    image

  • 先安装对应版本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

posted @ 2023-01-11 09:53  zt001  阅读(384)  评论(0编辑  收藏  举报