有时临时需要在笔记本上面使用TensorFlow,与Ubuntu平台上安装流程大同小异。
一、材料:
- 对应的显卡驱动
- CUDA & cudnn
- Anaconda(或miniconda)
二、步骤:
1、查看自己电脑对应的最新显卡驱动版本+>这里<+
2、确认显卡是否支持cuda加速+>这里<+
3、cuda下载链接& cudnn6.0下载链接[ps. TensorFlow1.3及以上最低支持到cudnn6.0],cuda正常安装后,将cudnn解压内容(三个文件)分别拷贝到对应的CUDA/v8.0对应目录下
4、 安装Anaconda或miniconda+>这里<+
Anaconda则是一个打包的集合,里面预装好了conda、某个版本的python、各种包、科学计算工具等。Miniconda,只包含最基本的内容——python与conda,以及相关的必须依赖项,对于资源受限的用户,Miniconda是一种选择(2.7/3.5版本可选)。
5、一旦Anaconda(miniconda)安装完成后就可以使用咱们的conda命令了,conda可以理解为一个工具,也是一个可执行命令,其核心功能是包管理与环境管理。包管理与pip的使用类似,环境管理则允许用户方便地安装不同版本的python并可以快速切换。
-
pip更换到清华源(避免被墙导致的装不上库,另外有时conda找不到的包可以用pip安装)
windows下,直接在user目录中创建一个pip目录,如:C:\Users\xx\pip,新建文件pip.ini
写入以下两行
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple -
创建一个独立的tensorflow环境(默认会在Anaconda/miniconda的envs的目录下生成你所命名的文件夹,python版本根据自己需求,我选的python3.5)
>conda create -n tensorflow python=3.5
>activate tensorflow
安装TensorFlow,下面安装方式将会安装最新版本的tensorflow,安装其他版本需要指定.whl相应连接,不加赘述;
--gpu版本
>pip install tensorflow-gpu
--cpu版本
>pip install tensorflow
验证一下,import tensorflow不报错就OK了
>python
>>>import tensorflow
捕获.PNG
-
如果需要opencv这些包,试一下(注意opencv版本)
>conda install --channel https://conda.anaconda.org/menpo opencv3
网友评论