使用環境
ubuntu22.04
nvidia-driber 525.60.11
CUDA 12
GPU 1650
CPU i7 9750H
メモリー 32GB
ノートPCです。
自分のまとめ用です
nvidia driver install ubuntu22.04&
Nouveauドライバーの停止
lsmod | grep nouveau
無効にするために、以下のコマンドを実行します。
sudo sh -c "echo 'blacklist nouveau' > /etc/modprobe.d/blacklist-nouveau.conf"
sudo sh -c "echo 'options nouveau modeset=0' >> /etc/modprobe.d/blacklist-nouveau.conf"
sudo update-initramfs -u
sudo reboot
再起動後、再度
lsmod | grep nouveau
を実行し、何も出力されないことを確認します。
lsmod | grep nouveau
NVIDIAドライバーのインストール
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
インストールするバージョンの検索
ubuntu-drivers devices
-openと付いているドライバーでは無いドライバーをインストール
sudo apt install -y nvidia-driver-525
sudo reboot
nvidia-smi
手動インストールの場合はdaemonが起動します。
詳しくは前回の記事を見てください。
https://qiita.com/wd01x2oa/items/7a4f513dba430cc205b9
gitがインストールされていない場合はインストールします。
atpとpipを更新します。
sudo apt update
sudo apt upgrade
sudo apt install python3-pip
sudo apt install git
機械学習ライブラリ PyTorchのインストール
pip3 install torch torchvision torchaudio
nvidia-driverのverとCUDAのver
自分が使っているOSを選んでください。
pytporchが入らない場合
pip install -U pip
pytorchの確認
python
Python 3.8.10 (default, May 26 2023, 14:05:08)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
import torch
torch.version
'2.0.1+cu117'
OpenCVをインストールとアンインストールとopencvのver確認
pip3 install opencv-python インストール
pip3 uninstall opencv-python アンインストール
open cv インストール
pip3 install opencv-python==4.5.1 これだとエラーが 出ます
このエラーが出たときは、opencvのverがわからないというエラーなのでfrom versionsの中から
入れたいverを選んでください。今回は4.6.0.66を入れたいので下記のコマンドを入力をしてください。
今回はopencvのpip3 install opencv-python==4.6.0.66です。
yolov5を新規で使用する時にGitHubの以下のリポジトリをDLします
git clone https://github.com/ultralytics/yolov5
yolov5を使用するために必要なライブラリのリストをインストールします。
cd yolov5
pip3 install -r requirements.txt
下のように表示されたらインストールは完了です。
Successfully installed MarkupSafe-2.1.1 Pillow-9.3.0 absl-py-1.3.0 asttokens-2.1.0 backcall-0.2.0 cachetools-5.2.0 charset-normalizer-2.1.1 contourpy-1.0.6 cycler-0.11.0 decorator-5.1.1 executing-1.2.0 fonttools-4.38.0 google-auth-2.14.1 google-auth-oauthlib-0.4.6 grpcio-1.50.0 importlib-metadata-5.0.0 ipython-8.6.0 jedi-0.18.1 kiwisolver-1.4.4 markdown-3.4.1 matplotlib-3.6.2 matplotlib-inline-0.1.6 packaging-21.3 pandas-1.5.1 parso-0.8.3 pickleshare-0.7.5 prompt-toolkit-3.0.32 protobuf-3.20.3 psutil-5.9.4 pure-eval-0.2.2 pyasn1-0.4.8 pyasn1-modules-0.2.8 pygments-2.13.0 pyparsing-3.0.9 python-dateutil-2.8.2 pytz-2022.6 requests-2.28.1 requests-oauthlib-1.3.1 rsa-4.9 scipy-1.9.3 seaborn-0.12.1 stack-data-0.6.1 tensorboard-2.11.0 tensorboard-data-server-0.6.1 tensorboard-plugin-wit-1.8.1 thop-0.1.1.post2209072238 tqdm-4.64.1 traitlets-5.5.0 wcwidth-0.2.5 werkzeug-2.2.2 zipp-3.10.0
#このようなエラーが出たら
ERROR: launchpadlib 1.10.13 requires testresources, which is not installed.
ERROR: pandas 2.0.3 has requirement python-dateutil>=2.8.2, but you'll have python-dateutil 2.7.3 which is incompatible.
python-dateutil>=2.8.2にpython-dateutil 2.7.3が互換性がないといういみなので
このコマンドを入力します。
pip install python-dateutil --upgrade
PyTorchとYOLOv5の動作確認を行います。
ターミナルで以下のコマンドを実行してください。
python3
import torch
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
以下のように表示されたら正常にPyTorchとYOLOv5が動作しています。
Fusing layers...
YOLOv5s summary: 213 layers, 7225885 parameters, 0 gradients
Adding AutoShape...
yolov5 start
cd ~/yolov5
~/yolov5$ python3 detect.py --source 0 #web camera
~/yolov5$ python3 detect3.py --source 'https://www.youtube.com/watch?v=zhe29G0vE0c' # youtube
yolov8 install
git clone https://github.com/ultralytics/ultralytics
pip install ultralytics
cd ultralytics
pip install -r requirements.txt
yolov8 start
cd ~/ultralytics
yolo task=detect mode=predict model=yolov8n.pt source= 0
#web camera
cd ~/ultralytics
yolo task=segment mode=predict model=yolov8x-seg.pt source='https://www.youtube.com/watch?v=U0ZejuhTSbY' #youtube
参考記事
https://qiita.com/porizou1/items/74d8264d6381ee2941bd
# Autowareとyoloの動画です
https://youtube.com/playlist?list=PLXccx3X6SIKwE5ty1FSp8yuAdz4rP8TRS&si=siuQilgPDrC_zG46