オムライスの備忘録

数学・統計学・機械学習・プログラミングに関することを記す

2020-08-13から1日間の記事一覧

「could not select device driver "" with capabilities: [[gpu]].」エラー

2020/08/13 環境 Ubuntu 16.4 docker 19.03.12 cuda 10.2 参考 Docker(19.03)でgpu有効化してpytorchで訓練するまでやる(Ubuntu18.04) DockerでGPUを使う 対応 nvidia-container-runtime-script.shを起動して、nvidia-container-runtime をインストールする…

tensorflow v.2.x で tensorboard in keras

2020/08/13 参考 TensorBoardスカラー:Kerasでのトレーニング指標のロギング tensorflow 公式 「カスタムスカラーのロギング」を参考 file_writer を作成して、tf.summary.xxx で書き込む # writer を作成 file_writer = tf.summary.create_file_writer(log…