「張先生」カテゴリーアーカイブ

週報(大塚)

GPUサーバーのコンテナにPyTorchをインストールする際に少しハマったのでメモを残します。

PyTorchはインストールできたけど、GPUが使えないって人は以下を参考にしてください。

・PyTorch 1.13 からCUDA 10.2 がdeprecatedになったらしい。

PyTorch 1.13 release, including beta versions of functorch and improved support for Apple’s new M1 chips.

・そのため、GET STARTEDからインストールしようとしてもCUDA 10.2が選べない。

・現在のGPUサーバーのCUDAのバージョンは10.2で、そのバージョンのcudatoolkitが入ってないとおそらく動かない。

・コンテナに新しいCUDAをインストールしようとするとコンテナが壊れるみたいなのでやめた方が良い。

今週の進捗(新野)

使用デバイス: cuda:0
-----start-------
Epoch 1/14 | train | Loss: 0.5769 Acc: 0.6843
Epoch 1/14 | val | Loss: 0.4572 Acc: 0.8090
Epoch 2/14 | train | Loss: 0.3569 Acc: 0.8503
Epoch 2/14 | val | Loss: 0.4530 Acc: 0.8256
Epoch 3/14 … Continue Reading ››

B4ゼミ

藤本:野菜写真データの水増し法を確認。YOLUV5のアノテーション学習セット確認

根来:セグメンテーションプログラム確認。マンホールの写真データしゅうしゅうし、クラス追加して学習を行う。

西:パソコンやスマホ画面の視認性向上のための色弱者対応色変更プロフィールの定義を行う。画像強調も視野。

Stay Hungry, Stay Foolish!