openkylin/pytorch repository overview

⁠Quick reference

• The official PyTorch docker image.

• Maintained by: openKylin RISC-V SIG⁠.

• Where to get help: openKylin RISC-V SIG⁠, openKylin⁠.

⁠PyTorch | openKylin

Current PyTorch docker images are built on the openKylin⁠. This repository is free to use and exempted from per-user rate limits.

PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.

Learn more about on PyTorch Website⁠.

⁠Supported tags and respective Dockerfile links

The tag of each PyTorch docker image is consist of the complete software stack version. The details are as follows

⁠Usage

1. Ensure that you have Docker installed, or are using Docker for Linux containers if on Windows.

2. Obtain the PyTorch ecosystem docker image. There are two ways to do this:

• Pull the pre-built Docker image

  # From DockerHub
  docker pull openkylin/pytorch
  # From Quay
  docker pull quay.io/openkylin/pytorch
  

  Copy

• Clone the source repository⁠⁠. Navigate to the AI/pytorch folder and build the image locally with the following command.

  docker build . -t openkylin/pytorch

1. Run the Docker container to launch the PyTorch developer image.

• docker run -it openkylin/pytorch

⁠Question and answering

If you have any questions or want to use some special features, please submit an issue or a pull request on openkylin-docker-images⁠.