最もホットな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) 真