CUDA 10.1 TensorFlow 2.xの下のWin10実行

最もホットな2つのフレームワークの深さの研究はpytorchとtensorflow、pytorch最新バージョンは、最新バージョンは1.3要件が10.1であるpytorch下のwin10バージョンCUDAで最高、以下を参照してください、2.0であるtensorflow、1.3です。

 

 

 

 下に示すように、CUDAのtensorflow 2.0とバージョン10.0を使用しました:

 

これは通常、CUDA 10.1インストール、紛争が生じ、その後、tensorflow 2.0 CUDA 10.1で実行するためにソースコードを再コンパイルしており、より困難なだけでなく、失敗に非常になりやすいが、今よりよい解決策があり、ソースコードやtensorflowをコンパイルする手順、

これは、インストールに直接毎晩バージョン、最新バージョンのtensorflow 2.1.0-RC2、使用ピップをインストールされています

ピップインストールtensorflow-GPU == 2.1.0-RC2

それをテスト

TFとしてインポートtensorflow
印刷(tf.test.is_gpu_available())

通常の使用のGPUことを発見しました

2020年1月1日10:40:26.616452:I tensorflow /コア/ common_runtime / GPU / gpu_device.cc:1241】作成TensorFlowデバイス(/デバイス:GPU:4702メガバイトのメモリ0) - >物理GPU(デバイス:0、名前:のGeForce GTX 1060、PCIバスID:0000:01:00.0、計算機能:6.1)

  

おすすめ

転載: www.cnblogs.com/lly277365/p/12128476.html