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__}")
반응형