CUDA_VISIBLE_DEVICES 안먹힐때 해결법! (GPU 할당하기)

아늬.. CUDA_VISIBLE_DEVICES = "0, 1"이런식으로 할당해도 작동이 안되는 것 아니겠는가..

출처 - https://discuss.pytorch.org/t/cuda-visible-device-is-of-no-use/10018/10

 

그래서 검색해봤더니, 이런 결과가 딱!!

 

해결법 - 

import torch 보다 더 전에 CUDA 관련 os.environ 코드들을 써주면 해결된다고 한다.

 

+

근데 웃긴건 내 상황은 이게 문제가 아니였네?! ㅋㅋ

이거 해도 안되길래 확인해봤더니 애초에 도커에서 GPU설정을 잘못해뒀었다..

 

설정을 바꾸고 나니, 위의 방법대로 안해도 잘된다..

정상적으로 세팅을 했는데도 안되시는 분들이 한번쯤은 해볼 수 있는 방법인듯

 

 

 

++

 

다른 컴퓨터에 세팅했다고 동일 문제 발생 -> 찾아보니, gpu랑 맞는 버전의 pytorch 버전이 필요한듯 하다..