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

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

  • https://blog.csdn.net/qq_34087914/article/details/108593764
  • https://tensorflow.google.cn/install/source_windows
  • https://www.jianshu.com/p/aafd88033867
  • https://blog.csdn.net/chao_1083934282/article/details/102802771
  • https://blog.csdn.net/zzpong/article/details/80282814
  • https://blog.csdn.net/qq_34606546/article/details/86538508

环境: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

你可能感兴趣的:(【深度学习TensorFlow(五)】GTX960M win10 安装GPU版的TensorFlow1.12)