当前位置: 代码迷 >> 综合 >> win10+anaconda+pychram+YOLO开发环境搭建
  详细解决方案

win10+anaconda+pychram+YOLO开发环境搭建

热度:86   发布时间:2023-12-14 03:02:40.0

想在win10上面训练网络的小伙伴看过来,之前找了很多教程发现都是这里问题,那里问题,今天就开始一次性解决吧,只需要很简单的几步。

首先

(1)下载一个anaconda 

下载完成之后

(2)打开这个网站

https://www.tensorflow.org/install/source_windows

看到下面的这个表这个是tf的版本对于的环境 

如果你想装tf2.0.0 你就需要cudnn7.4+cuda10

接下来我们

(3)去找cudnn7.4+cuda10

这个是cuda10的https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

直接下载得到安装包安装

下载得到cudnn7.4.1.5

全部安装后

(4)进入anaconda虚拟环境

直接

pip install tensorflow-gpu==2.0.0

(5)打开python

输入

import tensorflow as tftf.__vervison__

(6)下载pychram

https://www.jetbrains.com/pycharm/download/#section=windows

(7)创建新项目

(8)测试是否能用

测试代码:

import tensorflow as tfimport tensorflow as tftensorflow_version =tf.__version__#以下两行代码适合有布置gpu环境的
gpu_available = tf.test.is_gpu_available()print("tensorflow version:",tensorflow_version,"\tGPU available:",gpu_available)a = tf.constant([1.0,2.0],name="a")
b = tf.constant([1.0,2.0],name="b")result = tf.add(a,b,name="add")print(result)

 

 

(9)关于pychram汉化

 

选择Settings

找到Plugins 

输入Chinese>安装>重启使用

 

(10)避坑

1)关于PyCharm ImportError: No module named tensorflow

解决方案:

anaconda上面输入代码查看虚拟环境

conda info --env

复制虚拟环境的位置放到下面的位置

再次运行代码,成功。

 

 

  相关解决方案