tensorflow-gpu

发表于 | 更新于
| 本文字数： 1.4k

如何在win10上装tensorflow-gpu

简要介绍

我的MX150不支持CUDA 9.0，所以用了CUDA 9.2，而且还得用别人编译好的tensorflow

Anaconda

  • 环境变量
    • C:\Users\fq\Anaconda3
    • C:\Users\fq\Anaconda3\Library\mingw-w64\bin
    • C:\Users\fq\Anaconda3\Library\usr\bin
    • C:\Users\fq\Anaconda3\Library\bin
    • C:\Users\fq\Anaconda3\Scripts
  • 相关命令
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    # 配置清华镜像库
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --set show_channel_urls yes

    # 创建环境
    conda create -n tf python=3.6

    # 激活环境
    activate tf

    # 删除环境
    conda env remove -n tf

    # 升级pip
    python -m pip install --upgrade pip

    # 安装numpy、matplotlib、scikit-learn
    pip install numpy
    pip install matplotlib
    pip install pandas
    pip install scikit-learn

用 conda install ~ 安装相应的包时会有些问题

tensorflow

  • 下载得到tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl
  • 在cmd中进入tf环境pip install C:\dev\python_wheel\tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl
  • 检验
1
2
3
4
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

keras

1
pip install keras

可能出现的错误

  • 1.NVIDIA显卡驱动需要更新（我的版本是398.75）

Internal: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version；

  • 2.vscode中的设置

Linter pylint is not installed

解决：在设置中把Python › Linting: Enabled把√去了

0%