NVIDIA Docker Image list

참고
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

• Cuda 11.7, Ubuntu 20.04 (develop)
  https://hub.docker.com/layers/nvidia/cuda/11.7.0-devel-ubuntu20.04/images/sha256-e5b0bb8d257ed720b9749e69bae6888764355df1023bdd38a7906fbf55c882c1?context=explore

• Cuda 11.7, Ubuntu 22.04 (develop)
  https://hub.docker.com/layers/nvidia/cuda/11.7.1-devel-ubuntu22.04/images/sha256-5f711addc3f20a0a2120738be907a38d80a38d8fb567c8cfaa72af8719254570?context=explore

• Cuda 11.2, Ubuntu 20.04 (develop)
  https://hub.docker.com/layers/nvidia/cuda/11.2.2-devel-ubuntu20.04/images/sha256-2316d1439c91b82df73a0e9f9958ef1e63c437ff92c00763d411386a8db31973?context=explore

 

 

 

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

 

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