도커에 쿠다 설치

https://hub.docker.com/r/nvidia/cuda/tags

https://hub.docker.com/r/nvidia/cuda/tags

여기서 cuda 환경을 도커위에 세팅하기 위할 것이다. 

 

나는 권장되는 cuda가 12.6이므로 이를 설치하겠다.

 

cuda는 여기서 설치하자

https://hub.docker.com/r/nvidia/cuda/tags?page=&page_size=&ordering=&name=12.6

https://hub.docker.com/r/nvidia/cuda/tags?name=12.6&ordering=&page=&page_size=

페이지에서 맞는 걸 찾은 뒤 복사하여 아래처럼 붙여넣는다.

 

 

 

설치하면

docker run --gpus all -it nvidia/cuda:12.6.0-cudnn-devel-ubuntu20.04 /bin/bash

 

이 명령으로 이미지를 실행한다.

 

 

cuda 12.6버전이 설치된것을 볼 수 있다.