Segmentation학습 > Onnx > C# - (1)설치

참조사이트 : https://blog.hbsmith.io/c-%EA%B8%B0%EB%B0%98-semantic-segmentation-%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%A8-%EA%B0%9C%EB%B0%9C-d330a98a005b

 

* CUDA & CUDNN 환경

* CUDA : 11.3
* CUDNN : 8.2.1

1. 다음의 명령어로 가상환경을 만든다

conda create -n segmentation python=3.9
conda activate segmentation

2. github로부터 다운 받는다

git clone https://github.com/qubvel/segmentation_models.pytorch

3. 생성된 폴더로 이동한 뒤 필요한 녀석들을 설치한다.

cd segmentation_models.pytorch
pip install -r requirements.txt

4. GPU 버전을 설치하기 위해 다음의 항목들을 지우고 다시 설치한다.

(https://pytorch.org/ 참조)

pip uninstall torch
pip uninstall torchvision

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

5. 추가로 필요한 기능들을 설치한다

pip install matplotlib
pip install pytorch-lightning

6. Command Line 상에서 다음과 같이 가상환경 위치를 찾아낸다.

7. 가상환경의 경로에  path.pth 라는 텍스트 파일을 만든 뒤 github로부터 다운받은 루트폴더 경로를 저장한다

* 파일명 : d:\Anaconda3\envs\segmentation\path.pth
* 내용 --------------------------------------------------
D:\DeepLearning\Repository\segmentation_models.pytorch
---------------------------------------------------------