Open koljanich opened 1 year ago
One more: os Windows 10. CUDA and tensorflow installed by instruction https://www.tensorflow.org/install/pip?hl=en#windows-native. Verification python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))" returns [PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
Do you use an mirror image to install the library? If you use a Tsinghua image to install, the PyTorch you installed is a CPU version.
Please go to the official website to download the corresponding version.
I ran into a similar problem using Windows 11 with a NVIDIA GeForce RTX 3080 Ti. Even though the checks would show my GPU, it would only use CPU when processing a video. I ran the nvidia-smi.exe
in a command prompt and it showed I was using cuda version 12.2. At this time Pytorch only supports up to version 12.1. To resolve this, I downgraded my driver to 12.1 and it works using the GPU.
These steps assume you have Anaconda installed.
nvidia-smi.exe
and confirm you CUDA version. (I am using CUDA Version 12.1. My drive version is 531.14.)conda install cuda --channel nvidia/label/cuda-12.1.0
conda create --name imageai python=3.9
conda activate imageai
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
pip install -r requirements_gpu.txt
pip install imageai --upgrade
Thanks to Nilotpal Sinha PhD. His article on Medium helped me figure this out. - How to setup PyTorch with CUDA in Windows 11
Please how do we set that up on MacOs?
Hello! I try to use yolov3.pt tiny-yolov3.pt and retinanet_resnet50_fpn_coco-eeacb38b.pth. But result is same. CPU Usage 89%, GPU on minimal 18%. I use 1660 Super GPU from NVIDIA. Script: `# -- coding: utf-8 -- import cv2 import os from imageai.Detection import ObjectDetection
camera = cv2.VideoCapture(1) execution_path = os.getcwd()
detector = ObjectDetection() detector.setModelTypeAsTinyYOLOv3() detector.setModelPath(os.path.join(execution_path, "tiny-yolov3.pt")) detector.loadModel()
while True:
camera.release() cv2.destroyAllWindows()`