TensorFlowバージョンのアップグレード/GPUとCUDAバージョンの競合の解決

1036 ワード

現在tensorflowが公式に提供しているバージョンはcuda 9.0のみをサポートしており、非ソースコードのインストールにはGPUコンピューティング能力が3.5以上必要です.
注意公式に提供されているインストール方法では、最新のTensorFlow安定版がデフォルトでインストールされ、cuda 8.0をインストールしても実行できません.
C:\> pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade tensorflow-gpu

GPUの比較的古い機械に対して、2つの方法が解決します
1)ソースコードからのコンパイル、ソースコードのコンパイルはwindowsをサポートしていませんが、ドキュメントは強引なコンパイルの解決策を提供しているようです
2)TensorFlowバージョンを下げ、1.4.0バージョン(cuda 8.0)は3.0 GPUをサポートする
pip install tensorflow-gpu==1.4.0

注意、pythonのpackageをアップグレードするには注意してください.バージョンが衝突する可能性があります.大丈夫です.アップグレードをお勧めしません.
——————————————————————
TensorFlowのアップグレード:
pip install --upgrade tensorflow-gpu
pip install--upgrade tensorflow

現在のtensorflowバージョンを表示するには、次の手順に従います.
python
import tensorflow as tf
tf.__version__

クエリーtensorflowインストールパスは次のとおりです.
tf.__path__

Ref:
https://blog.csdn.net/junmuzi/article/details/78357371