windows安装tensorflow2的CPU和GPU版本

安装anaconda(以python3.7.0为例)https://blog.csdn.net/u010916338/article/details/81168668安装虚环境(1) 查看所有python虚环境conda info –envs# conda environments:#base * C:\ProgramData\Anaconda3(2) 安装python虚环境# 注:tf-gpu-260是虚环境的名称conda cre..

  1. 安装anaconda(以python3.7.0为例)
    /参考博客地址

  2. 安装虚环境

(1) 查看所有虚环境

conda info --envs

# conda environments:
#
base                  *  C:\ProgramData\Anaconda3

(2) 安装虚环境

# 注:tf-gpu-260是虚环境的名称
conda create --name tf-gpu-260 python=3.7.0

(3) 进入虚环境

activate tf-gpu-260

(4) 退出虚环境

deactivate
或者
conda deactivate

(5) 删除虚环境

conda remove --name tf-gpu-260 --all
  1. 在虚环境中安装tensorflow

(1)进入虚环境

activate tf-gpu-260

(2)查看本机支持哪些tensorflow版本

conda search tensorflow-gpu
或者
conda search tensorflow

windows安装tensorflow2的CPU和GPU版本
(3)查看tensorflow各个版本依赖的环境
查看版本依赖
windows安装tensorflow2的CPU和GPU版本
windows安装tensorflow2的CPU和GPU版本
(4)安装tensorflow(注:安装GPU版本系统会自动根据机器环境安装相应的CUDA和cuDNN)

# 安装gpu2.6版本
conda install tensorflow-gpu==2.6.0
# 安装cpu2.6版本
conda install tensorflow==2.6.0

(5)在虚环境中进入python环境测试tensorflow-GPU是否安装成功

import tensorflow as tf
gpu_device_name = tf.test.gpu_device_name()
print(gpu_device_name)
tf.test.is_gpu_available()

#  /device:GPU:0
#  True
  1. 使用虚环境

(1)在原始环境中安装插件和ipykernel(在jupyter中可以选择环境)

conda install nb_conda
conda install ipykernel

(2)虚环境中安装ipykernel

activate tf-gpu-260
conda install ipykernel

注:一般会使用Matplotlib包,虚环境中没有这个包,需要手动安装!!!

# 因为安装ipykerner时conda会自动升级,再使用conda安装其它包时可能会报错,建议使用pip命令安装
pip install matplotlib

windows安装tensorflow2的CPU和GPU版本
windows安装tensorflow2的CPU和GPU版本

版权声明:本文为博主醉糊涂仙原创文章,版权归属原作者,如果侵权,请联系我们删除!

原文链接:https://blog.csdn.net/u010916338/article/details/122567377

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
社会演员多的头像社会演员多普通用户
上一篇 2022年1月19日
下一篇 2022年1月19日

相关推荐