当前位置:   article > 正文

梳理一下TensorFlow的环境配置和Tensorflow-gpu的环境配置_配置tensorflow环境

配置tensorflow环境

一.Tensorflow2.6环境配置

1.创建并激活虚拟环境

conda create -n tensorflow2.6 python==3.8

conda activate tensorflow2.6

2.安装相关软件包

pip清华镜像源:https://pypi.tuna.tsinghua.edu.cn/simple
华为云镜像源:https://mirrors.huaweicloud.com/repository/pypi/simple/
阿里云镜像源: https://mirrors.aliyun.com/pypi/simple/

pip install numpy matplotlib Pillow scikit-learn pandas -i  https://mirrors.aliyun.com/pypi/simple/

 3.安装tensorflow2.6

pip install tensorflow==2.6.0 -i https://mirrors.aliyun.com/pypi/simple/

4.命令行输入python,打开交互模式,输入import tensorflow

1.这里报错说matplotlib和pandas版本和numpy不兼容的话见我的另外一篇博客

安装TensorFlow2.6遇到的问题-CSDN博客


2.如果是import tensorflow 出现两行内容(虽然这是tensorflow-CPU,现在默认好像下载是GPU版本,所以缺少显卡加速)

>>> import tensorflow as tf

2023-04-02 12:12:57.522370: W tensorflow/stream_executor/platform/default/http://dso_loader.cc:60] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found

2023-04-02 12:12:57.541160: I tensorflow/stream_executor/cuda/http://cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine. 

解决办法如下:


 1.因为一会还要配置GPU,当把下面GPU配置好后上面这个问题缺少显卡加速问题也就解决了

在 Windows 环境中从源代码构建  |  TensorFlow (google.cn)icon-default.png?t=N7T8https://tensorflow.google.cn/install/source_windows?hl=zh-cn#gpu(TensorFlow的官网地址查看版本对应号)


2. 下载CUDA官网网址CUDA Toolkit Archive | NVIDIA Developericon-default.png?t=N7T8https://developer.nvidia.com/cuda-toolkit-archive

下载11.2版本根据自己电脑情况进行选择。


 3.cuDNN下载官网cuDNN Archive | NVIDIA Developer

下载如图所示的cuDNN8.1版本(选择一个),注明:这里需要注册登录才能进行下载。 


 二.CUDA和cuDNN配置(因为我已经配置过了,找了图片可以按照下面的做)

这里有四个选项时把多余的那个取消勾选

 

后面保持默认安装即可安装完毕后

nvcc -V

 

 2.1cuDNN配置

1.新建一个文件夹cuDNN将其解压至新建文件夹下


 2.找到cuDNN文件下这三个文件


 3.找到CUDA安装位置默认为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2 


 4.分别将cuDNN文件下的bin、inclue、lib文件添加到CUDA的 bin、inclue、lib文件下

注明:添加不是替换昂

5.配置PATH系统环境变量并将上移到第三位置

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\CUPTI\lib64

 配置完如上:import tensorflow的问题也解决了。


三.Tensorflow2.6-gpu环境配置

1.创建并激活虚拟环境

conda create -n tensorflow-gpu python==3.8

conda activate tensorflow-gpu

2.安装相关软件包

pip清华镜像源:https://pypi.tuna.tsinghua.edu.cn/simple
华为云镜像源:https://mirrors.huaweicloud.com/repository/pypi/simple/
阿里云镜像源: https://mirrors.aliyun.com/pypi/simple/

pip install numpy matplotlib Pillow scikit-learn pandas -i  https://mirrors.aliyun.com/pypi/simple/

 3.安装tensorflow-gpu

pip install tensorflow-gpu==2.6.0 -i https://mirrors.aliyun.com/pypi/simple/

4.命令行输入python,打开交互模式,输入import tensorflow

 5.如若出现版本冲突依旧是

安装TensorFlow2.6遇到的问题-CSDN博客icon-default.png?t=N7T8https://blog.csdn.net/qq_62331166/article/details/133618955?spm=1001.2014.3001.5501

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/盐析白兔/article/detail/695673
推荐阅读
相关标签
  

闽ICP备14008679号