Docker

NVIDIA Docker Image list

scone 2023. 3. 2. 18:54
참고
Pytorch 공식 Cuda ver. 11.7 is the latest

검색 출처
Docker hub uploaded by Nvidia
https://hub.docker.com/r/nvidia/cuda/tags?page=2&name=11.2.2
 

NVIDIA Docker Image

 

 

 

또는 Pytorch 계정으로 업로드 된 Pytorch가 깔린 버전의 이미지를 아래의 링크에서 검색할 수 있습니다.
그러나, 이는 최신 버전인 Cuda 11.7를 반영하고 있지 않으며, 원하는 Ubuntu 버전으로 커스터마이징 하기 어려울 수 있기 때문에 위에 적은 이미지를 가지고 Pytorch 또는 Tensorflow를 직접 까는 것이 더 나아 보입니다.

 

https://hub.docker.com/r/pytorch/pytorch/tags