반응형
시작전
docker-ce는 설치되어 있어야 함
gpg 키와 저장소 추가
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
&& curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
&& curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
apt package repository update
$ sudo apt update
nvidia-docker2 설치
$ sudo apt install nvidia-docker2
필요시 테스트
$ docker run --rm --gpus all ubuntu:18.04 nvidia-smi
반응형
'개발, 웹, 블로그 > DeepLearning 상식' 카테고리의 다른 글
간단히 docker 환경에서 ollama를 이용하여 llama3.1 8B 실행해보기 (0) | 2024.10.23 |
---|---|
docker: Error response from daemon: unknown or invalid runtime name: nvidia. 오류 (0) | 2024.09.25 |
Python ONNX Runtime으로 Resnet 테스트 하기 (0) | 2024.02.08 |