Search

240905_1757_h100 PC에 torch, einops, timm, matplotlib, tensorboard 세팅 [1]

아래 명령어로 PyTorch를 CUDA 11.8과 함께 설치

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
10분 소요
정상적으로 동작되는지 확인
(d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP# python -c "import torch; print(torch.cuda.is_available())" True (d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP#
JavaScript
복사

ModuleNotFoundError: No module named 'einops'

(d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP# python main.py -k cpn_ft_h36m_dbb -c checkpoint -gpu 0 --nolog --evaluate h36m_best_epoch.bin -num_pro Traceback (most recent call last): File "main.py", line 21, in <module> from einops import rearrange, repeat ModuleNotFoundError: No module named 'einops' (d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP# pip install einops Collecting einops Downloading einops-0.8.0-py3-none-any.whl.metadata (12 kB) Downloading einops-0.8.0-py3-none-any.whl (43 kB) Installing collected packages: einops Successfully installed einops-0.8.0 WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager, possibly rendering your system unusable.It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv. Use the --root-user-action option if you know what you are doing and want to suppress this warning.
JavaScript
복사

ModuleNotFoundError: No module named 'timm'

(d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP# python main.py -k cpn_ft_h36m_dbb -c checkpoint -gpu 0 --nolog --evaluate h36m_best_epoch.bin -num_pro Traceback (most recent call last): File "main.py", line 27, in <module> from common.diffusionpose import * File "/mnt/host/data/glory/240905_1736_D3DP/D3DP/common/diffusionpose.py", line 18, in <module> from common.mixste import * File "/mnt/host/data/glory/240905_1736_D3DP/D3DP/common/mixste.py", line 18, in <module> from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD ModuleNotFoundError: No module named 'timm' (d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP# pip install timm Collecting timm Downloading timm-1.0.9-py3-none-any.whl.metadata (42 kB) Requirement already satisfied: torch in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from timm) (2.4.1) Requirement already satisfied: torchvision in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from timm) (0.19.1) Requirement already satisfied: pyyaml in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from timm) (6.0.1) Collecting huggingface_hub (from timm) Using cached huggingface_hub-0.24.6-py3-none-any.whl.metadata (13 kB) Collecting safetensors (from timm) Downloading safetensors-0.4.4-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (3.8 kB) Requirement already satisfied: filelock in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from huggingface_hub->timm) (3.13.1) Collecting fsspec>=2023.5.0 (from huggingface_hub->timm) Downloading fsspec-2024.9.0-py3-none-any.whl.metadata (11 kB) Collecting packaging>=20.9 (from huggingface_hub->timm) Using cached packaging-24.1-py3-none-any.whl.metadata (3.2 kB) Requirement already satisfied: requests in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from huggingface_hub->timm) (2.32.3) Collecting tqdm>=4.42.1 (from huggingface_hub->timm) Downloading tqdm-4.66.5-py3-none-any.whl.metadata (57 kB) Requirement already satisfied: typing-extensions>=3.7.4.3 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from huggingface_hub->timm) (4.11.0) Requirement already satisfied: sympy in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from torch->timm) (1.13.2) Requirement already satisfied: networkx in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from torch->timm) (3.1) Requirement already satisfied: jinja2 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from torch->timm) (3.1.4) Requirement already satisfied: numpy in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from torchvision->timm) (1.24.3) Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from torchvision->timm) (10.4.0) Requirement already satisfied: MarkupSafe>=2.0 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from jinja2->torch->timm) (2.1.3) Requirement already satisfied: charset-normalizer<4,>=2 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from requests->huggingface_hub->timm) (3.3.2) Requirement already satisfied: idna<4,>=2.5 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from requests->huggingface_hub->timm) (3.7) Requirement already satisfied: urllib3<3,>=1.21.1 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from requests->huggingface_hub->timm) (2.2.2) Requirement already satisfied: certifi>=2017.4.17 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from requests->huggingface_hub->timm) (2024.7.4) Requirement already satisfied: mpmath<1.4,>=1.1.0 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from sympy->torch->timm) (1.3.0) Downloading timm-1.0.9-py3-none-any.whl (2.3 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.3/2.3 MB 20.6 MB/s eta 0:00:00 Using cached huggingface_hub-0.24.6-py3-none-any.whl (417 kB) Downloading safetensors-0.4.4-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (436 kB) Downloading fsspec-2024.9.0-py3-none-any.whl (179 kB) Using cached packaging-24.1-py3-none-any.whl (53 kB) Downloading tqdm-4.66.5-py3-none-any.whl (78 kB) Installing collected packages: tqdm, safetensors, packaging, fsspec, huggingface_hub, timm Successfully installed fsspec-2024.9.0 huggingface_hub-0.24.6 packaging-24.1 safetensors-0.4.4 timm-1.0.9 tqdm-4.66.5 WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager, possibly rendering your system unusable.It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv. Use the --root-user-action option if you know what you are doing and want to suppress this warning. (d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP#
JavaScript
복사

ModuleNotFoundError: No module named 'matplotlib'

(d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP# python main.py -k cpn_ft_h36m_dbb -c checkpoint -gpu 0 --nolog --evaluate h36m_best_epoch.bin -num_pro Traceback (most recent call last): File "main.py", line 29, in <module> from common.loss import * File "/mnt/host/data/glory/240905_1736_D3DP/D3DP/common/loss.py", line 1, in <module> from matplotlib.pyplot import bone ModuleNotFoundError: No module named 'matplotlib' (d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP# pip install matplotlib Collecting matplotlib Downloading matplotlib-3.7.5-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.whl.metadata (5.7 kB) Collecting contourpy>=1.0.1 (from matplotlib) Downloading contourpy-1.1.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (5.9 kB) Collecting cycler>=0.10 (from matplotlib) Using cached cycler-0.12.1-py3-none-any.whl.metadata (3.8 kB) Collecting fonttools>=4.22.0 (from matplotlib) Downloading fonttools-4.53.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (162 kB) Collecting kiwisolver>=1.0.1 (from matplotlib) Downloading kiwisolver-1.4.7-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.whl.metadata (6.3 kB) Requirement already satisfied: numpy<2,>=1.20 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from matplotlib) (1.24.3) Requirement already satisfied: packaging>=20.0 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from matplotlib) (24.1) Requirement already satisfied: pillow>=6.2.0 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from matplotlib) (10.4.0) Collecting pyparsing>=2.3.1 (from matplotlib) Using cached pyparsing-3.1.4-py3-none-any.whl.metadata (5.1 kB) Collecting python-dateutil>=2.7 (from matplotlib) Using cached python_dateutil-2.9.0.post0-py2.py3-none-any.whl.metadata (8.4 kB) Collecting importlib-resources>=3.2.0 (from matplotlib) Using cached importlib_resources-6.4.4-py3-none-any.whl.metadata (4.0 kB) Collecting zipp>=3.1.0 (from importlib-resources>=3.2.0->matplotlib) Using cached zipp-3.20.1-py3-none-any.whl.metadata (3.7 kB) Collecting six>=1.5 (from python-dateutil>=2.7->matplotlib) Using cached six-1.16.0-py2.py3-none-any.whl.metadata (1.8 kB) Downloading matplotlib-3.7.5-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (9.2 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 9.2/9.2 MB 49.4 MB/s eta 0:00:00 Downloading contourpy-1.1.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (301 kB) Using cached cycler-0.12.1-py3-none-any.whl (8.3 kB) Downloading fonttools-4.53.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.7 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 4.7/4.7 MB 93.0 MB/s eta 0:00:00 Using cached importlib_resources-6.4.4-py3-none-any.whl (35 kB) Downloading kiwisolver-1.4.7-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.whl (1.2 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.2/1.2 MB 145.4 MB/s eta 0:00:00 Using cached pyparsing-3.1.4-py3-none-any.whl (104 kB) Using cached python_dateutil-2.9.0.post0-py2.py3-none-any.whl (229 kB) Using cached six-1.16.0-py2.py3-none-any.whl (11 kB) Using cached zipp-3.20.1-py3-none-any.whl (9.0 kB) Installing collected packages: zipp, six, pyparsing, kiwisolver, fonttools, cycler, contourpy, python-dateutil, importlib-resources, matplotlib Successfully installed contourpy-1.1.1 cycler-0.12.1 fonttools-4.53.1 importlib-resources-6.4.4 kiwisolver-1.4.7 matplotlib-3.7.5 pyparsing-3.1.4 python-dateutil-2.9.0.post0 six-1.16.0 zipp-3.20.1 WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager, possibly rendering your system unusable.It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv. Use the --root-user-action option if you know what you are doing and want to suppress this warning. (d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP#
JavaScript
복사

ModuleNotFoundError: No module named 'tensorboard'

(d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP# python main.py -k cpn_ft_h36m_dbb -c checkpoint -gpu 0 --nolog --evaluate h36m_best_epoch.bin -num_pro Traceback (most recent call last): File "main.py", line 34, in <module> from torch.utils.tensorboard import SummaryWriter File "/root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages/torch/utils/tensorboard/__init__.py", line 1, in <module> import tensorboard ModuleNotFoundError: No module named 'tensorboard' (d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP# pip install tensorboard Collecting tensorboard Downloading tensorboard-2.14.0-py3-none-any.whl.metadata (1.8 kB) Collecting absl-py>=0.4 (from tensorboard) Downloading absl_py-2.1.0-py3-none-any.whl.metadata (2.3 kB) Collecting grpcio>=1.48.2 (from tensorboard) Downloading grpcio-1.66.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (3.9 kB) Collecting google-auth<3,>=1.6.3 (from tensorboard) Downloading google_auth-2.34.0-py2.py3-none-any.whl.metadata (4.7 kB) Collecting google-auth-oauthlib<1.1,>=0.5 (from tensorboard) Downloading google_auth_oauthlib-1.0.0-py2.py3-none-any.whl.metadata (2.7 kB) Collecting markdown>=2.6.8 (from tensorboard) Downloading Markdown-3.7-py3-none-any.whl.metadata (7.0 kB) Requirement already satisfied: numpy>=1.12.0 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from tensorboard) (1.24.3) Collecting protobuf>=3.19.6 (from tensorboard) Downloading protobuf-5.28.0-cp38-abi3-manylinux2014_x86_64.whl.metadata (592 bytes) Requirement already satisfied: requests<3,>=2.21.0 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from tensorboard) (2.32.3) Requirement already satisfied: setuptools>=41.0.0 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from tensorboard) (72.1.0) Collecting tensorboard-data-server<0.8.0,>=0.7.0 (from tensorboard) Downloading tensorboard_data_server-0.7.2-py3-none-manylinux_2_31_x86_64.whl.metadata (1.1 kB) Collecting werkzeug>=1.0.1 (from tensorboard) Downloading werkzeug-3.0.4-py3-none-any.whl.metadata (3.7 kB) Requirement already satisfied: wheel>=0.26 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from tensorboard) (0.43.0) Collecting cachetools<6.0,>=2.0.0 (from google-auth<3,>=1.6.3->tensorboard) Downloading cachetools-5.5.0-py3-none-any.whl.metadata (5.3 kB) Collecting pyasn1-modules>=0.2.1 (from google-auth<3,>=1.6.3->tensorboard) Downloading pyasn1_modules-0.4.0-py3-none-any.whl.metadata (3.4 kB) Collecting rsa<5,>=3.1.4 (from google-auth<3,>=1.6.3->tensorboard) Downloading rsa-4.9-py3-none-any.whl.metadata (4.2 kB) Collecting requests-oauthlib>=0.7.0 (from google-auth-oauthlib<1.1,>=0.5->tensorboard) Downloading requests_oauthlib-2.0.0-py2.py3-none-any.whl.metadata (11 kB) Collecting importlib-metadata>=4.4 (from markdown>=2.6.8->tensorboard) Downloading importlib_metadata-8.4.0-py3-none-any.whl.metadata (4.7 kB) Requirement already satisfied: charset-normalizer<4,>=2 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from requests<3,>=2.21.0->tensorboard) (3.3.2) Requirement already satisfied: idna<4,>=2.5 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from requests<3,>=2.21.0->tensorboard) (3.7) Requirement already satisfied: urllib3<3,>=1.21.1 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from requests<3,>=2.21.0->tensorboard) (2.2.2) Requirement already satisfied: certifi>=2017.4.17 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from requests<3,>=2.21.0->tensorboard) (2024.7.4) Requirement already satisfied: MarkupSafe>=2.1.1 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from werkzeug>=1.0.1->tensorboard) (2.1.3) Requirement already satisfied: zipp>=0.5 in /root/miniconda3/envs/d3dp_env/lib/python3.8/site-packages (from importlib-metadata>=4.4->markdown>=2.6.8->tensorboard) (3.20.1) Collecting pyasn1<0.7.0,>=0.4.6 (from pyasn1-modules>=0.2.1->google-auth<3,>=1.6.3->tensorboard) Downloading pyasn1-0.6.0-py2.py3-none-any.whl.metadata (8.3 kB) Collecting oauthlib>=3.0.0 (from requests-oauthlib>=0.7.0->google-auth-oauthlib<1.1,>=0.5->tensorboard) Downloading oauthlib-3.2.2-py3-none-any.whl.metadata (7.5 kB) Downloading tensorboard-2.14.0-py3-none-any.whl (5.5 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 5.5/5.5 MB 35.7 MB/s eta 0:00:00 Downloading absl_py-2.1.0-py3-none-any.whl (133 kB) Downloading google_auth-2.34.0-py2.py3-none-any.whl (200 kB) Downloading google_auth_oauthlib-1.0.0-py2.py3-none-any.whl (18 kB) Downloading grpcio-1.66.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (5.8 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 5.8/5.8 MB 105.5 MB/s eta 0:00:00 Downloading Markdown-3.7-py3-none-any.whl (106 kB) Downloading protobuf-5.28.0-cp38-abi3-manylinux2014_x86_64.whl (316 kB) Downloading tensorboard_data_server-0.7.2-py3-none-manylinux_2_31_x86_64.whl (6.6 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 6.6/6.6 MB 79.0 MB/s eta 0:00:00 Downloading werkzeug-3.0.4-py3-none-any.whl (227 kB) Downloading cachetools-5.5.0-py3-none-any.whl (9.5 kB) Downloading importlib_metadata-8.4.0-py3-none-any.whl (26 kB) Downloading pyasn1_modules-0.4.0-py3-none-any.whl (181 kB) Downloading requests_oauthlib-2.0.0-py2.py3-none-any.whl (24 kB) Downloading rsa-4.9-py3-none-any.whl (34 kB) Downloading oauthlib-3.2.2-py3-none-any.whl (151 kB) Downloading pyasn1-0.6.0-py2.py3-none-any.whl (85 kB) Installing collected packages: werkzeug, tensorboard-data-server, pyasn1, protobuf, oauthlib, importlib-metadata, grpcio, cachetools, absl-py, rsa, requests-oauthlib, pyasn1-modules, markdown, google-auth, google-auth-oauthlib, tensorboard Successfully installed absl-py-2.1.0 cachetools-5.5.0 google-auth-2.34.0 google-auth-oauthlib-1.0.0 grpcio-1.66.1 importlib-metadata-8.4.0 markdown-3.7 oauthlib-3.2.2 protobuf-5.28.0 pyasn1-0.6.0 pyasn1-modules-0.4.0 requests-oauthlib-2.0.0 rsa-4.9 tensorboard-2.14.0 tensorboard-data-server-0.7.2 werkzeug-3.0.4 WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager, possibly rendering your system unusable.It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv. Use the --root-user-action option if you know what you are doing and want to suppress this warning. (d3dp_env) root@bm-h100-11:/mnt/host/data/glory/240905_1736_D3DP/D3DP#
JavaScript
복사

안녕하세요

한국전자기술연구원 김영광입니다.
관련 기술 문의와 R&D 공동 연구 사업 관련 문의는 “glory@keti.re.kr”로 연락 부탁드립니다.

Hello

I'm Yeonggwang Kim from the Korea Electronics Research Institute.
For technical and business inquiries, please contact me at “glory@keti.re.kr”