PyTorchをWin10のGPUで走らせる
tensorflow-gpu2.0.0でkerasを使う
tensorflow2.0.0がリリースされたので試してみた。スクリプト冒頭のkerasのインポート宣言を変えるだけで動くようだ。
参考
infohub Developmentさんの『TensorFlowに組み込まれたKerasを使う方法』
環境
Windows10 Home バージョン1903
Python3.6.6
NVIDIAR GeForce GTX1050T…
tensorflowの環境更新：Win10にKeras2.2.5、tensorflow-gpu14.1.0、CUDA10.0、cuDNN7.63、NVIDIAドライバ436.30を入れた
Generative Deep Learningを体験するため、Windows10でkeras+tensorflow-gpuの環境を最新版に更新した。結果CUDA８をCUDA10に入れ替え、NVIDIAのドライバーも新しいものとした。以下はその記録。
参考
・@konzo_さんのブログ、『NvidiaドライバとCUDAとcuDNNとTensorflow-gpuとPythonのバージョ…
