web-dev-qa-db-fra.com

Comment installer la version CPU de tensorflow en utilisant conda

Si je lance conda install tensorflow conda veut installer la version GPU, avec CUDA etc. Je n'ai pas de GPU Nvidia donc je veux installer la version CPU uniquement.

$ conda install tensorflow
Solving environment: done

## Package Plan ##

  environment location: /home/cjk/.conda/envs/athion-forecaster

  added / updated specs: 
    - tensorflow


The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    tensorflow-base-1.12.0     |gpu_py36had579c0_0       216.9 MB
    tensorflow-1.12.0          |gpu_py36he74679b_0           3 KB
    _tflow_select-2.1.0        |              gpu           2 KB
    tensorboard-1.12.0         |   py36hf484d3e_0         3.1 MB
    ------------------------------------------------------------
                                           Total:       220.1 MB

The following NEW packages will be INSTALLED:

    _tflow_select:       2.1.0-gpu                
    cudatoolkit:         9.2-0                    
    cudnn:               7.2.1-cuda9.2_0          
    cupti:               9.2.148-0                
    keras-applications:  1.0.6-py36_0             
    keras-preprocessing: 1.0.5-py36_0             
    tensorboard:         1.12.0-py36hf484d3e_0    
    tensorflow:          1.12.0-gpu_py36he74679b_0
    tensorflow-base:     1.12.0-gpu_py36had579c0_0

Proceed ([y]/n)?

Fonctionnement conda search tensorflow montre que les packages CPU (en fait deux types, eigen et mkl du dernier tensorflow existent:

conda search tensorflow
...
tensorflow               1.12.0 eigen_py27hfe19c55_0  pkgs/main           
tensorflow               1.12.0 eigen_py36hbd5f568_0  pkgs/main           
tensorflow               1.12.0 gpu_py27h2a0f108_0  pkgs/main           
tensorflow               1.12.0 gpu_py27h956c076_0  pkgs/main           
tensorflow               1.12.0 gpu_py36he68c306_0  pkgs/main           
tensorflow               1.12.0 gpu_py36he74679b_0  pkgs/main           
tensorflow               1.12.0 mkl_py27hc55d17a_0  pkgs/main           
tensorflow               1.12.0 mkl_py36h69b6ba0_0  pkgs/main

Comment dire à conda d'installer la version MKL?

J'ai essayé conda install tensorflow[mkl] mais cela donne les mêmes résultats que sans [mkl]. conda install tensorflow=mkl échoue avec PackagesNotFoundError.

tensorflowcondaintel-mkl
7
user7813790

Je l'ai trouvé: conda install 'tensorflow=*=mkl*'

10
user7813790

conda install tensorflow-mkl a fonctionné pour moi, pas la version avec les astérisques.

0
Egor Kraev