search

Hzzone / pytorch-openpose

pytorch implementation of openpose including Hand and Body Pose Estimation.
2.02k stars 391 forks source link

Environment Setup (running succesful!) #81

Open WhaSukGO opened 11 months ago

WhaSukGO commented 11 months ago

Tested on nvcr.io/nvidia/cuda:11.2.0-cudnn8-devel-ubuntu20.04

nvidia-smi

$ conda create -n pytorch-openpose-py38 python=3.8
$ conda activate pytorch-openpose-py38
$ conda install -y pytorch">=1.6.0" torchvision cudatoolkit">=9.2" -c pytorch
$ pip3 install -r requirements.txt
$ python3 demo.py

result

chrissunny94 commented 8 months ago 
  File "demo_camera.py", line 15, in <module>
    print(f"Torch device: {torch.cuda.get_device_name()}")
  File "/root/anaconda3/envs/pytorch-openpose-py38/lib/python3.8/site-packages/torch/cuda/__init__.py", line 419, in get_device_name
    return get_device_properties(device).name
  File "/root/anaconda3/envs/pytorch-openpose-py38/lib/python3.8/site-packages/torch/cuda/__init__.py", line 449, in get_device_properties
    _lazy_init()  # will define _get_device_properties
  File "/root/anaconda3/envs/pytorch-openpose-py38/lib/python3.8/site-packages/torch/cuda/__init__.py", line 289, in _lazy_init
    raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
chrissunny94 commented 8 months ago

NVIDIA-SMI

Sun Oct 15 11:28:13 2023 

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.199.02   Driver Version: 470.199.02   CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  Off  | 00000000:01:00.0  On |                  N/A |
| N/A   53C    P5    N/A /  N/A |     71MiB /  4040MiB |     30%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      2127      G   /usr/lib/xorg/Xorg                 69MiB |
+-----------------------------------------------------------------------------+