2021-12-30から1日間の記事一覧

KerasによるGPUメモリの占有を防ぐ

teratail.com 以下を実行する。 import os os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'

Tensorflow(Keras)でGPUを使って計算する

1. nvidia-cuda-toolkitのインストール sudo apt install nvidia-cuda-toolkitlibcuda v10.1がダウンロードされた $ ls /usr/lib/x86_64-linux-gnu/libcuda* /usr/lib/x86_64-linux-gnu/libcudart.so.10.1環境変数に以下を追加 export PATH=/usr/lib/x86_64-…