Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
Adrian Wälchli 49ed2b102b | 2 weeks ago | |
---|---|---|
.. | ||
base-cuda | 2 weeks ago | |
docs | 7 months ago | |
nvidia | 5 months ago | |
release | 3 weeks ago | |
README.md | 3 months ago |
You can build it on your own, note it takes lots of time, be prepared.
git clone https://github.com/Lightning-AI/lightning.git
# build with the default arguments
docker image build -t pytorch-lightning:latest -f dockers/base-cuda/Dockerfile .
# build with specific arguments
docker image build -t pytorch-lightning:base-cuda-py3.9-torch1.13-cuda11.7.1 -f dockers/base-cuda/Dockerfile --build-arg PYTHON_VERSION=3.9 --build-arg PYTORCH_VERSION=1.13 --build-arg CUDA_VERSION=11.7.1 .
To run your docker use
docker image list
docker run --rm -it pytorch-lightning:latest bash
and if you do not need it anymore, just clean it:
docker image list
docker image rm pytorch-lightning:latest
To run docker image with access to your GPUs, you need to install
# Add the package repositories
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker
and later run the docker image with --gpus all
. For example,
docker run --rm -it --gpus all pytorchlightning/pytorch_lightning:base-cuda-py3.9-torch1.13-cuda11.7.1
docker image build -t pytorch-lightning:v1.6.5 -f dockers/nvidia/Dockerfile --build-arg LIGHTNING_VERSION=1.6.5 .
docker run --rm -it --gpus=all -p 8888:8888 pytorch-lightning:v1.6.5
http://hostname:8888/?token=0719fa7e1729778b0cec363541a608d5003e26d4910983c6
hostname
by localhost
No Description
Python YAML TSX Dockerfile Shell other
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》