The minimum required Cuda capability is 3.5

前端 未结 3 1117
礼貌的吻别
礼貌的吻别 2021-02-07 04:00

After installing TensorFlow and its dependencies on a g2.2xlarge EC2 instance I tried to run an MNIST example from the getting started page:

python tensorflow/m         


        
3条回答
  •  挽巷
    挽巷 (楼主)
    2021-02-07 04:22

    There is a section in the official installation page that guides you to enable Cuda 3, but you need to build Tensorflow from source.

    $ TF_UNOFFICIAL_SETTING=1 ./configure
    
    # Same as the official settings above
    
    WARNING: You are configuring unofficial settings in TensorFlow. Because some
    external libraries are not backward compatible, these settings are largely
    untested and unsupported.
    
    Please specify a list of comma-separated Cuda compute capabilities you want to
    build with. You can find the compute capability of your device at:
    https://developer.nvidia.com/cuda-gpus.
    Please note that each additional compute capability significantly increases
    your build time and binary size. [Default is: "3.5,5.2"]: 3.0
    
    Setting up Cuda include
    Setting up Cuda lib64
    Setting up Cuda bin
    Setting up Cuda nvvm
    Configuration finished
    

提交回复
热议问题