Torch 2.11.0 on OpenCloudOS 9
基本信息
- 框架版本：v2.11.0
- 基础镜像：opencloudos9-cuda-devel:12.8
- Python 版本：3.11
- CUDA 版本： 12.x 或 更高
docker build -t oc9-pytorch:2.11.0 .
镜像启动命令
docker run -d --gpus all --name oc9-pytorch oc9-pytorch:2.11.0
镜像测试命令
docker run --rm --gpus all oc9-pytorch:2.11.0 bash test.sh