Tensorflow import error

前端 未结 2 975
误落风尘
误落风尘 2021-01-02 08:21

I\'m trying to install tensorflow which supports GPU.

I tried the information in the following link

https://www.tensorflow.org/install/install_windows

<
2条回答
  •  青春惊慌失措
    2021-01-02 08:58

    Please download the proper drivers as latest version of cuda v8.0 toolkit needs tensorflow 1.4.x needs.

    As, by default pip3 install --upgrade tensorflow-gpu install latest version of tensorflow which needs cuda v9.0 (latest) not cuda v8.0 toolkit

    You'll need to install the version 1.4.1 for CUDA-8 as

    pip install tensorflow-gpu==1.4.1
    

    The latest (version 1.5) is for CUDA-9 You'll need to install the version 1.5 for CUDA-9 as

    pip install tensorflow-gpu==1.5
    

    Download cuddn for your appropriate cuda version toolkit from here and follow this.

    Download appropriate cuda toolkit from here

提交回复
热议问题