The minimum required Cuda capability is 3.5

前端 未结 3 1119
礼貌的吻别
礼貌的吻别 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 simple trick. You don't even have to build TF from sources.

    In the file tensorflow\python\_pywrap_tensorflow.pyd there are two occurences of regex 3\.5.*5\.2. Just replace both 3.5 with 3.0.

    Tested on Windows 10, Anaconda 4.2.13, Python 3.5.2, TensorFlow 0.12, CUDA 8, NVidia GTX 660m (CUDA cap. 3.0).

提交回复
热议问题