nuvic/pytorch repository overview

# docker run --gpus 指定物理GPU  -it --name 自定义容器名称 镜像名称:版本(相应版本要求物理机nvidia driver支持) bash
docker run --network host --gpus '"device=0,1"' -it --name pytorch nuvic/pytorch:2.2.0-python3.10-cuda11.2.2-ubuntu20.04 bash

import torch

print(torch.cuda.is_available())  # 返回True表明正常

# docker run --gpus 指定物理GPU  -it --name 自定义容器名称 镜像名称:版本(相应版本要求物理机nvidia driver支持) bash
docker run --network host --gpus '"device=0,1"' -it --name pytorch nuvic/miniconda:24.1.2-cuda11.2.2-ubuntu20.04 bash

conda install python=3.10
conda install pytorch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=11.8 -c pytorch -c nvidia

import torch

print(torch.cuda.is_available())

conda clean -y --all && exit

# docker commit -m "说明" 容器名称 镜像名称
docker commit -m "pytorch is available." pytorch nuvic/pytorch:2.2.0-python3.10-cuda11.2.2-ubuntu20.04

# docker login
docker push nuvic/pytorch:2.2.0-python3.10-cuda11.2.2-ubuntu20.04