docker pytorch 2.1.1, cuda 12.1, python 3.10

pytorch/pytorch:2.1.1-cuda12.1-cudnn8-devel

 

 

https://hub.docker.com/layers/pytorch/pytorch/2.1.1-cuda12.1-cudnn8-devel/images/sha256-b3d31523ed2f8cfe3f0af826bfccc8836c119cb5a06bc0c86ae82e611e1317e0?context=explore

https://hub.docker.com/layers/pytorch/pytorch/2.1.1-cuda12.1-cudnn8-devel/images/sha256-b3d31523ed2f8cfe3f0af826bfccc8836c119cb5a06bc0c86ae82e611e1317e0?context=explore

 

1) docker pull

 

docker pull pytorch/pytorch:2.1.1-cuda12.1-cudnn8-devel

 

 

2) docker run 

 

docker run -it --gpus all --restart always --name pytorch2.1.1 -v /data:/data pytorch/pytorch:2.1.1-cuda12.1-cudnn8-devel /bin/bash

 

확인

 

3) 레포복사