神经网络学习(一)(傻瓜版)Tensorflow-GPU安装及Tensorflow Objection Detection API环境搭建
搞了好长时间一直出现各种问题,不是版本不对应就是显存不够,好在发现了一种非常香的安装方式。之前安装各种依赖包都是使用pip方式,后来发现各种版本问题(cudaToolkit跟cudnn)。这次所有依赖包都是使用conda 下载,conda会自动选定正确版本的依赖包!
一、 Tensorflow-GPU安装
先说一下我的电脑DELL-G3 GTX1650Ti,tensorflow-gpu==1.14.0。
-
安装Anaconda3,安装时选择自动添加环境变量。
-
打开Anaconda Prompt
-
先创建虚拟环境 ,在Anaconda Prompt里键入
conda create --name 这里写环境名称 我的是tensorflow python=3.6
此次我选的是Python3.6,因为接下来我要安装tensorflow-gpu==1.14.0,若选择安装tensorflow高版本这里建议选择Python3.7。
-
激活所创建的虚拟环境
键入 activate tensorflow(这里的tensorflow是我上面自己起的名,你的可以是tensorflow-gpu或是别的 都是可以的。)若是激活成功 在行前括号里的内容会变成虚拟环境名称(我的是tensorflow)如下图
-
傻瓜式安装tensorflow
键入conda install tensorflow-gpu=1.14.0
这里一定用conda。然后就等吧 这里下载的很慢甚至会连接失败,解决方法两种:1.科学上网 2.改清华源如下(键入如下三行代码):
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
- 验证是否安装成功
进入python(在Anaconda Prompt里 先激活虚拟环境,然后键入python
后回车 然后键入import tensorflow as tf
tf.test.is_gpu_available()
)如下图:
若是返回True 则安装成功。如下图:
二、Tensorflow Objection Detection API环境搭建
过两天再写,待续。。。。。。。。