【深度学习TensorFlow(五)】GTX960M win10 安装GPU版的TensorFlow1.12

学习自:- https://mp.weixin.qq.com/s/E7wL2UgsJ3qAEK1cTZWatQ

环境:win10
显卡:英伟达RTX960M

一、安装anaconda

https://www.anaconda.com

二、NVIDIA驱动程序的安装

Windows 环境中,如果系统具有 NVIDIA 显卡,则往往已经自动安装了 NVIDIA 显卡驱动程序。如未安装,直接访问 NVIDIA 官方网站 下载并安装对应型号的最新公版驱动程序。

三、CUDA Toolkit 和 cnDNN 的安装

用anaconda新建并进入conda环境



3.1 查看cuda版本对应的版本信息

进入英伟达显卡的控制面板



3.2 查看tensorflow、python、cuda、cudnn版本对应关系:


更改下载源:

### 3.3 更改源,加快下载速度

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

3.4 下载库:

python -m pip install --upgrade pip
#conda install cudatoolkit=9.0
#conda install cudnn=7.0
conda install tensorflow-gpu=1.12 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
pip3 install pillow

继续安装:

pip install -U keras-tuner -i https://pypi.douban.com/simple/
conda install matplotlib
pip install opencv-python

四、检查

(DL) PS C:\Users\Robin> python
Python 3.6.2 |Continuum Analytics, Inc.| (default, Jul 20 2017, 12:30:02) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> tf.__version__
'1.12.0'

实际测试


image.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容