GPU #Device 썸네일형 리스트형 [GPU]GPU 셋팅 시 버전 체 1. GPU가 사용 가능한 상태인지 확인 import os os.environ['CUDA_DEVICE_ORDER']="PCI_BUS_ID" os.environ['CUDA_VISIBLE_DEVICES']='0' device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') print('Device:', device) # print('Current cuda device:', torch.cuda.current_device()) print('Count of using GPUs:', torch.cuda.device_count()) [Out] 2. 해당 서버 또는 컨테이너의 shell에서 CUDA버전을 확인 CUDA버전과 torch관련 라이브러리의 버전.. 더보기 이전 1 다음
