[pytorch] 파이토치 opencv, mxnet, torchmetrics 설치 시 gpu 인식 불가 이슈 해결

pytorch나 tensorflow를 다루다보면 어떠한 패키지를 설치했을 때, 패키지가 cpu 버전으로 내려가는 이슈가 있어서 gpu 인식을 못한다. 

 

이때 해결하기가 굉장히 번거로운데, 필자 같은 경우에는 가상환경을 처음부터 재설치했다. 

 

이런 이슈는 특히 텐서플로우보다는 pytorch에서 더 많이 발생했다.

 

그중에서도 많이 설치하는 라이브러리를 위주로 몇개 설명을 하자면...

 

opencv

이 패키지는 왜인지는 모르겠지만 conda를 통해 opencv를 설치했을 때는 pytorch에서 cuda를 인식하지 못하게 되었다.

이 경우 pip install opencv-python 명령어를 통해 설치하면 말끔하게 해결되니 참고하자.

 

mxnet

mxnet과 같은 경우도 정상적으로 설치하면 문제가 되지 않으나.. 나처럼 gpu 인식에 노이로제가 걸린 사람처럼 정보를 찾다가 설치하면 문제가 될 수 있다.

 

pip install mxnet 으로 설치를 진행한다면 gpu 인식에 아무 문제 없이 설치가 진행된다. (O)

 

하지만 conda 커맨드를 통해 (py-mxnet) 을 설치하는 경우 torch에서 cuda를 인식하지 못하는 이슈가 발생하게 된다.

conda install -c anaconda py-mxnet  (X)

 

 

torchmetrics

토치메트릭스 같은 경우는 pytorch를 설치한 경우에는 문제가 되지 않으나...

torch를 설치한 경우에는 gpu 인식이 되지 않는다. 이점 유의하자.

 

아래 코드를 참고하자

# gpu 인식에 영향을 안주는 설치방법 pytorch 명시
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

pip install torchmetrics  # 이후 torchmetrics 설치 진행


# gpu 인식에 영향을 줌 (인식 못하게 하는 이슈가 생김!!)
# 아래처럼 torch를 설치하는 경우가 문제가 생김
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

# 이후 torchmetrics 설치하면 cuda 인식 불가
pip install torchmetrics

 

요점은

 

pytorch를 설치하고, torchmetrics를 설치하면 정상 작동한다 (O)

pytorch가 아닌 torch를 설치하고 torchmetrics를 설치하면 gpu 인식을 못하는 에러가 발생한다 (X)