tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode

摘要:
简介:在之前版本的Tensorflow1.10中,发现CNN的训练将自动停止。最后,当添加卷积层时,训练将崩溃/停止。只有使用完整的连接层才能实现正常的训练。我非常喜欢这个IDE。我建议你在墙上有裂缝时使用它。此步骤不是必需的。您还可以使用熟悉的IDE下载:https://code.visualstudio.com/安装后即可使用。您可以在左侧的插件列中找到您想要或不想要的各种python插件。3.CUDA10.0下载地址:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal现在据说只有基本的软件包,没有补丁。只需下载基本软件包并安装即可。

引言:

之前用的tensorflow 1.10版本,发现在训练CNN的时候会自动中止,最后定位到加入卷积层就会导致训练崩溃/中止,只用全连接层却能正常训练。重装一天后无果,干脆全部升级使用tensorflow1.15:

改用WIN10+python3.7+tensorflow1.15.0+CUDA10.0(+cudnn7.6.5)+VScode

顺便记录下过程

我的笔记本配置:ryzen7 4800h+RTX2060+16G 3200 DDR4 RAM+1.5T SSD 

因为我们要安装的GPU版本的tensorflow, 只要大佬你的显卡和我的一样(无论是笔记本还是台式机)就可以照这个流程安装了

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第1张

安装目录:

1.python3.7

2.VScode(或者各位大佬熟悉的其他IDE)

3.CUDA10.0

4.CUDNN7.6

5.tensorflow_gpu 1.15.0

1.python3.7

注意tensorflow_gpu 1.15.0对应必须使用python3.7的版本,如果大佬们原来用的其他版本,要先换为py3.7的环境

从win10自带应用市场安装最方便,而且环境自动配好

1.电脑桌面左下角win旁边的搜索框中输入 microsoft store, 打开应用商店后右上角找python搜索

2.从结果中找到python3.7安装

 tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第2张

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第3张

验证:如果大佬第一次装python, 可以粗略用cmd验证一下:cmd窗口输入python -V , 会提示python 版本 3.7.x

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第4张

进一步验证:可以打开桌面左下角搜索框输入IDLE  (python自带的IDE)

print('hello world')+回车(注意使用英文输入法), 没报错则okay

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第5张

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第6张

 2.IDE:VScode

全称为:Visual studio code . 这个IDE相对pycharm来说更简洁,有巨量的第三方插件,还自带的第三方插件安装平台,免费而且啥计算机语言都支持。很喜欢这个IDE,这里墙裂推荐各位大佬使用。这步不是必要的,各位大佬也可以使用自己熟悉的IDE

微软的官网下载:https://code.visualstudio.com/

安装既能用,在左方插件栏可以找到各种你想要或者不想要的python插件(其中一个叫python的插件必须安装不然编译不了)

 tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第7张

 

3.CUDA10.0

下载地址:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

现在说只有基础包,没有补丁,下载基础包安装就好了。

·安装的时候出现选择里边,驱动和pyhsx不是必须的(前提是如果你已经安装了比安装提示更新版本的驱动)

·安装里边VS2015相关的东西不是必须的,如果你只在python环境下做深度学习

·环境是安装的时候自动加好的

注意下载版本是对的,如下图:是CUDA Toolkit 10.0 ; 是CUDA Toolkit 10.0 ; 是CUDA Toolkit 10.0 ; 是CUDA Toolkit 10.0 ; 重要的事情说3遍

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第8张

验证:

桌面右下角搜索窗口输入cmd 

输入nvcc -V

会显示CUDA的信息

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第9张

4.CUDNN 7.6.5

官方下载链接(需要登陆,如果大佬第一次用,最快的时选择社交账号->微信登陆):https://developer.nvidia.com/rdp/cudnn-archive

选择 CUDNN v7.6.5 for CUDA 10.0 里边的 win10版本

注意:选择 CUDNN v7.6.5 for CUDA 10.0  ; 选择 CUDNN v7.6.5 for CUDA 10.0 ; 选择 CUDNN v7.6.5 for CUDA 10.0 重要的事情说三遍

 tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第10张

下载后解压,并把解压后的文件全部复制到CUDA中

我们在上一步装的是CUDA10.0 ,默认路径如下:

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第11张

5.tensorflow_gpu

我们要安装的是tensorflow 1 系列最新(现在来说)的版本1.15, 需求的CUDA10.0 和CUDNN我们已经在上面两步装上了

cmd窗口 , 用pip install 指令安装, 为了加速,这里我们使用国内清华镜像站加速:

完整的指令为:

pip install tensorflow_gpu==1.15.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

所有的依赖包都会顺带安装好:

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第12张

我在安装过程报了一次错:

ERROR: Could not install packages due to an EnvironmentError: [Errno 2] No such file or directory: 'C:\Users\Denn.Dumbledore\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.7_qbz5n2kfra8p0\LocalCache\local-packages\Python37\site-packages\tensorflow_estimator\python\estimator\canned\linear_optimizer\python\utils\__pycache__\sharded_mutable_dense_hashtable.cpython-37.pyc'

原因是注册列表长度限制:

找到一个网上的解答https://stackoverflow.com/questions/59388330/tensorflow-installation-problem-on-windows-10

引用h.meng的回答:HKEY_LOCAL_MACHINESYSTEMCurrentControlSetControlFileSystem@LongPathsEnabled to 1. You can do it by Win+R, and type regedit and you can change the setting above. Hope my answer works for you.

解除注册列表长度限制后,重新按照第5步安装一下

成功会提示succefully installed xxx包 xxx包 xxx包...

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第13张

 到这里所有的安装已经完成了

验证:

为了确保这次安装的tensorflow1.15没有出现之前1.10奇怪的训练问题,我这里用包含一个卷积层、一个池化层、一个全连接层的简单模型测试一下:

1.随便在哪个你喜欢的路径建立一个文件夹,在文件夹里边新建一个python文件随便命名,比如:Demo.py

2.打开VScode

3.把刚刚新建的文件夹用鼠标左键拖到vscode的主窗口中,输入下面这个模型的代码,按F5运行

注意:

·此Demo需要使用mnist数据集测测试:

(这个数据集是第一次运行自动从网络指定路径下载的,如果是各位大佬第一次使用这个数据集可能会出现下载数据集超时的网络问题,这样子的话可以直接报错链接处下载好数据,再放入keras dataset的路径,具体方法可以百度搜索错误内容)

·这个DEMO是一个3层模型:

  ·1个卷积层6*5*5

  ·1个池化层2*2

  ·1个全连接层120

展开浏览代码:

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第14张tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第15张
#Demo
import tensorflow as tf 
import numpy as np

#train set
(x_train,y_train),(x_test,y_test)=tf.keras.datasets.mnist.load_data()

#normalization
x_train = tf.keras.utils.normalize(x_train)
x_test = tf.keras.utils.normalize(x_test)

#one more channel is needed to convolution layer
x_train=x_train.reshape(-1,28,28,1)
x_test=x_test.reshape(-1,28,28,1)

#label:one_hot()
y_train = tf.keras.utils.to_categorical(y_train, 10)
y_test = tf.keras.utils.to_categorical(y_test, 10)#numpy


#model instantiation
model= tf.keras.models.Sequential()
#convolution  6*5*5 
model.add(tf.keras.layers.Conv2D(6,(5,5),activation='relu',input_shape=(28,28,1),padding="same"))
#pooling  2*2
model.add(tf.keras.layers.MaxPooling2D(pool_size=(2,2)))
#dropout, in case of over fitting
model.add(tf.keras.layers.Dropout(0.25))
#flatten
model.add(tf.keras.layers.Flatten())
#full connection 120
model.add(tf.keras.layers.Dense(120,activation='relu'))
#dropout, in case of over fitting
model.add(tf.keras.layers.Dropout(0.5))
#output layer : softmax 
model.add(tf.keras.layers.Dense(10,activation='softmax'))
#set optimizer and loss function
model.compile(optimizer='adam',loss=tf.keras.losses.categorical_crossentropy,metrics=['accuracy'])

#train
result = model.fit(x_train,y_train,epochs=2,validation_data=(x_test,y_test))
print('fitting process finished')

#get loss and accuratity
val_loss,val_acc=model.evaluate(x_test,y_test)
print('loss:',val_loss,'acc:', val_acc)

#check with the first 10 data in the test set
predictions=model.predict(x_test[:10])

print(predictions)
for i in range(10):
    predicted_number = np.argmax(predictions[i])#get the max one, which is the most probablity of the prediction
    print( 'prediction:',predicted_number,'groundTrue:',y_test[i])
View Code

输出结果:

tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode第16张

免责声明:文章转载自《tensorflow安装: win10 + RTX2060 + tensorflow1.15.0+ cuda10.0 + VScode》仅用于学习参考。如对内容有疑问,请及时联系本站处理。

上篇SQL Server 获取(本周、本月、本旬、本季、本年)的某一天Asterisk Manager Interface(AMI)下篇

宿迁高防,2C2G15M,22元/月;香港BGP,2C5G5M,25元/月 雨云优惠码:MjYwNzM=

相关文章

python(数字Number)

数字 Python 数字数据类型用于存储数值。 数据类型是不允许改变的,这就意味着如果改变数字数据类型的值,将重新分配内存空间 Python 支持三种不同的数值类型: 整型(int) - 通常被称为是整型或整数,是正或负整数,不带小数点。Python3 整型是没有限制大小的,可以当作 Long 类型使用,所以 Python3 没有 Python2 的 Lo...

Thinkphp6笔记二:开启多应用模式

开启多应用模式1.首先刪除app目录下所有其他文件2.安装多应用模式扩展: composer require topthink/think-multi-app3.创建三大应用index(前端),admin(后台),common(公共应用)命令:php think build index命令:php think build admin命令:php think...

接口测试中,json格式的报文发送时提示报文格式错误

最近在接口测试时遇到一个奇怪的问题,data参数为json格式,运行时总是提示报文格式不对,把data参数放到JSON在线解析里面报文也是正确的。 后来通过打印data找到了问题所在,在python中打印data数据,会自动把双引号变为单引号,而单引号JSON格式不认,所以提示报文格式错误。 可以使用json.dumps(data)方法将数据保持双引号...

java实现安全证书相关操作

https://blog.csdn.net/zhushanzhi/article/details/77864516 版权声明:本文为博主原创文章,未经博主允许不得转载。 [java]view plaincopy packagetest; importjava.io.ByteArrayOutputStream; importjava.io.File;...

python 最短路径

贾格尔(Jagger)找到一张地图,该地图指示大量宝藏的位置,并希望找到它们。 该地图将几个位置标记为节点和几个边缘,这表示两个位置直接相连。 总共有n个节点和m个边。 贾格尔(Jagger)位于节点1,宝物位于节点n。 当他运行最短路径算法以找出通往宝藏的最短路径时,他突然发现除了他的起始节点和宝藏的位置以外,每个节点都有一个怪物。 节点u上的怪物具有力...

OpenStack镜像制作笔记 以windows8.1amd64为例

by hyc 目录 1.下载win8_64位的iso文件 2.下载对应电脑的vnc 3.下载Xshell软件 4.连接成功后,在Xshell下安装软件包 5.下载FileZillaClient软件 6.驱动下载 7.镜像的安装与制作 8.用vnc远程安装系统 9.安装完后进行一些设置 9.1设置MTU 9.2关闭防火墙 9.3开启任何人远程访问 9.4更...