20180804 Anaconda+python3.6+Tensorflow-gpu1.9 环境搭建

20180804 Anaconda+python3.6+Tensorflow-gpu1.9 环境搭建

继昨天放弃了学习Rust之后,今天打算跟随舍友学习一下(玩一下)机器学习的经典框架Tensorflow,自己的笔记本电脑显卡是1050Ti的,自然而然要想到采用GPU加速啦。

根据网上搭建环境的理论,需要安装:

①Python3.6 à(pip)Tensorflow-gpu1.9

②Cuda9.0+Cudnn7

这两条线路不矛盾,可以同时下载……Python(推荐使用?)Anaconda虚拟环境与环境隔离开(为啥啊?)。

一、Anaconda+Python+Tensorflow安装

环境是win10,Anaconda下载之后就会自动安装python,也可以自行下载节省时间,并行操作。

这里,安装完python之后,首先要做的更改pip的更新源——不然网速特别难受。这里推荐使用豆瓣的更新源:

在”C:\Users\<用户名>\”下新建目录”pip”,在”pip”目录下建立文件”pip.ini”,文件的内容是:

[global]

trusted-host =  pypi.douban.com

index-url = http://pypi.douban.com/simple

原理估计是pip在检测更新源前会检测这个配置文件……网上一查一大把,也有清华、阿里云、网易等更新源,换了国内源速度就起飞了。

下一步是安装Tensorflow-gpu,可以先装着,与下面的Cuda安装不矛盾。

在Anaconda命令行中输入:

pip install tensorflow-gpu

(将自动安装目前源上有的最新版本1.9)

二、安装Cuda+Cudnn

去Nvidia官网下载Cuda9.0(其他版本的不行哈),链接:

https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

有1.33GB,贼大,去休息一下就下载完了。

顺便同时下载Cudnn7(需要注册Nvidia的账户,极其讨厌,网页有时候显示不正常)。然后我找同学复制了一个……230MB左右。

Cuda的安装方法:如果打得开,就直接安装,选自定义,

 

(估计这选的还多了,只要里面的dll文件就够了)

要是觉得慢,有个更粗暴的方法,这个Cuda的安装文件可以看作一个压缩包,用某国产压缩软件打开,并搜索dll,把所有的dll文件拷贝出来即可。

 

拷贝或者安装到一个文件夹就好了。

下一步是添加环境变量。将那些dll所在的文件夹对应的路径添加到系统的环境变量中即可——为了让tensorflow-gpu找得到那些文件。

(图略,基本操作)

留个疑问:

如果WSL的tensorflow-gpu在加载so文件的时候,能够把dll正确地链接上去,会不会也能正确运行呢?

目前WSL不能接触到硬件,希望微软能改进一下?我感觉WSL还是挺好玩的。

三、检测安装是否成功

继上述两个大的步骤完成以后,需要检测一下,打开Anaconda或者python命令行,输入代码:

>>> import tensorflow as tf

>>> hello = tf.constant('Hello, TensorFlow!')

>>> sess = tf.Session()

>>> sess.run(hello)

b’Hello, TensorFlow!’

显示上述结果说明安装完成。我的运行截图如下:(电脑好慢,估计又要放弃了)

 

猜你喜欢

转载自www.cnblogs.com/AmnesiaBeing/p/9418414.html
今日推荐