OpenMMLab(미공개)/MMEngine

[MMEngine] 환경 설정

huginn30 2024. 1. 14. 15:59
반응형

1. 가상환경 실행 및 pytorch 설치

- conda 가상환경 실행

conda create -n mmlab python=3.9
conda activate mmlab

 

- pytorch 설치 및 버전 확인

import torch
print(f"torch version: {torch.__version__}")
print(f"GPU: {torch.cuda.is_avalialble()}")

 

2. MMEngine 설치 및 확인

- MMEigine 설치 

pip install --force-reinstall mmengine

 

- MMEigine 설치 확인

import mmengine
print(f"mmengine version: {mmengine.__version__}")
반응형