[Docker] GPU 사용

yoonene·2023년 5월 8일
0

도커에서 GPU를 사용하려면 다음과 같이 run해야 한다.

docker run --runtime=nvidia -e NVIDIA_DRIVER_CAPABILITIES=compute,utility -e NVIDIA_VISIBLE_DEVICES=all {Docker Image}

profile
NLP Researcher / Information Retrieval / Search

0개의 댓글