Tensorflow安装方法(Win10、GPU Driver Version=399.24)
!先查看python版本再安装对应的Ancaconda!
1. 查看GPU驱动版本,及可对应的CUDA版本号
查看GPU驱动版本方式:
在C:\Program Files\NVIDIA Corporation\NVSMI目录下打开cmd,并输入:
C:\Program Files\NVIDIA Corporation\NVSMI> nvidia-smi
2. 查看CUDA版本号所对应的Tensorflow-GPU版本号,以及对应的Python版本
3. 安装与Python版本对应的Anaconda
4. 在Anaconda虚拟环境中安装Tensorflow-GPU
打开Anaconda Prompt
先配置清华镜像
(base) C:\Users\刘俊良> conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
(base) C:\Users\刘俊良> conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
(base) C:\Users\刘俊良> conda config --set show_channel_urls yes
创建虚拟环境(注意Python版本需对应Tensorflow-GPU)
(base) C:\Users\刘俊良> conda create -n Tensorflow python=3.6
(base) C:\Users\刘俊良> conda activate Tensorflow
安装Tensorflow-GPU
(Tensorflow) PS C:\Users\刘俊良> conda install tensorflow-gpu=1.8