spxiong/pytorch repository overview

Pure pytorch docker images with different os, cuda, and python versions.

Github: https://github.com/xiongsp/pytorch-docker⁠

Docker Hub: https://hub.docker.com/r/spxiong/pytorch⁠

⁠Changelog

• 2024/12/01: Support for pytorch 1.4-2.0.
• 2024/11/05: Support for pytorch 2.5.x. For 2.1-, please refer to this DockerHub⁠ and Github Repo⁠.
• 2024/10/10: Support for pytorch 2.1.x 2.2.x 2.3.x 2.4.x (I'm back).

⁠Usage

⁠Preparation

• Docker⁠
• Nvidia-docker⁠ (If a CUDA is required)

⁠Use pytorch docker

As same as Pytorch Official⁠.

docker run -it --rm \
    --gpus all \
    --net host
    -v /path/to/project:/path/to/project \
    -v /path/to/dataset:/path/to/dataset \
    spxiong/pytorch:[TAG]

Note: /path/to/project and /path/to/dataset is your own project path and dataset path, should be replaced in use.