import os
os.environ['CUDA_VISIBLE_DEVICES'] = '3'
이렇게 os로 gpu를 설정하는 것보다 import torch를 더 먼저 실행하여 생기는 문제
import torch를 하게 되면 기본적으로 0번 GPU를 torch가 가져온다고 한다.
▶os.environ을 먼저 실행하여 torch보다 먼저 GPU 권한을 설정한다. (그냥 무조건 os.environ부터 제일 먼저 실행하기)
+ 추가적으로 pytorch 환경변수가 꼬여서 생기는 오류이기도 하다.
맨 처음에 환경변수 해제 명령어를 실행해준다.
$ unset LD_LIBRARY_PATH
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '3'
import torch
torch.cuda.device_count()
'Setting, Error > Python, Python Library' 카테고리의 다른 글
Scikit-learn TimeSeriesSplit (0) | 2024.08.05 |
---|---|
[Error] no module named skimage (0) | 2024.06.18 |
Argparser 사용법 (0) | 2024.04.09 |
Pytorch 실행 에러 cuDNN version incompatibility (0) | 2024.03.26 |
[Error] WARNING: The script jupyter-console is installed in which is not on PATH. Consider adding this directory to PATH or, if you prefer to suppress this warning, use --no-warn-script-location. (0) | 2023.10.31 |