도커 이미지 -> #nsml: nvcr.io/nvidia/pytorch:19.06-py3
세션 만드는 명령어 -> nsml run -d tr-3 -g 1 --memory 24G -e main.py -a "mode test"
제출하는 명령어 -> nsml submit t0046/tr-3/104 checkpoint
-> nsml submit -t t0046/tr-3/103 checkpoint
현재 위와 같이 CPU만 잡히고 있습니다. GPU 잡히는 방법 알고 싶습니다.
저희 코드상에서도 GPU 사용하게 코드 해놓았는데 GPU가 안잡히고 있습니다. 로컬에서는 GPU로 잘 잡혀서 속도 확인 하였습니다.
도커 이미지 -> #nsml: nvcr.io/nvidia/pytorch:19.06-py3 세션 만드는 명령어 -> nsml run -d tr-3 -g 1 --memory 24G -e main.py -a "mode test" 제출하는 명령어 -> nsml submit t0046/tr-3/104 checkpoint -> nsml submit -t t0046/tr-3/103 checkpoint 현재 위와 같이 CPU만 잡히고 있습니다. GPU 잡히는 방법 알고 싶습니다. 저희 코드상에서도 GPU 사용하게 코드 해놓았는데 GPU가 안잡히고 있습니다. 로컬에서는 GPU로 잘 잡혀서 속도 확인 하였습니다.
nsml 상에서 submit 했을 때 GPU를 할당하는 방법을 알고 싶습니다.