当前位置:   article > 正文

envi深度学习环境_envi深度学习配置

envi深度学习配置


一、基本环境

Tensorflow与Python、CUDA、cuDNN的版本对应表_tensorflow版本对应python3.11_Ly.Leo的博客-CSDN博客

Windows安装tensorflow-gpu=1.14.0&CUDA=10.0&cuDNN=7.4 (多版本CUDA共存)_cuda10.0-CSDN博客

win10 查看GPU型号,驱动版本,CUDA版本_查看gpu版本-CSDN博客

二、安装cuda环境和cudnn

2.1安装cuda环境

2.2安装cudnn

三、安装tensorflow

打开DOS 命令窗口Win+R,输入cmd

3.1创建环境

conda create -n tensorflow-gpu1.14 python==3.6

3.2激活环境:

conda activate tensorflow-gpu1.14

3.3安装相关软件包

pip install包名

pip install numpy matplotlib Pillow scikit-learn pandas -I https://pypi.tuna.tsinghua.edu.cn/simple

出现错误

安装TensorFlow报错:No matching distribution found for tensorflow-CSDN博客

解决方法:

使用命令:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn 包名

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn numpy matplotlib Pillow scikit-learn pandas

阿里云:http://mirrors.aliyun.com/pypi/simple/

3.4安装 TensorFlow

3.4.1安装 TensorFlow

  1. pip install tensorflow==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
  2. pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ --upgrade tensorflow==2.0.0

3.4.2安装 TensorFlow-gpu

  1. pip install tensorflow-gpu==1.14.0
  2. pip install -i http://mirrors.aliyun.com/pypi/simple/--upgrade tensorflow-gpu==1.14.0
  3. pip3 install -i https://pypi.douban.com/simple/ --upgrade tensorflow-gpu==1.14.0

3.5测试GPU是否安装成功

激活python

输入测试语句:

  1. Import tensorflow as tf
  2. tf.__version__
  3. tf.test.is_gpu_available()
  4. tensorflow_test.is_gpu_available()

返回值为 True,则安装正确。

四、envi环境测试

测试出来有问题,想哭啦

*** Training a TensorFlow Mask Model ***
RuntimeError: A patchesPerBatch size of 3 is too large based on the available memory. Error: A patchesPerBatch size of 3 is too
     large based on the available memory.  Traceback (most recent call last):
ERROR: Model could not be trained.

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

闽ICP备14008679号