学习自:- 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
二、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'
实际测试