本项目为 pytorch 和 cuda 的基础镜像,可以作为基础docker进行相关AI项目的开发。
| 镜像地址 | Python版本 | Pytorch版本 | CUDA版本 |
|---|---|---|---|
| docker.cnb.cool/yichenyanyu/pytorch/pytorch2.6.0-cuda12.6-cudnn9-devel:latest | |||
| docker.cnb.cool/yichenyanyu/pytorch/pytorch2.68.0-cuda12.9-cudnn9-devel:latest |
根据需要在自己的项目 .cnb.yml 文件中引用当前项目的镜像,例如:
$: vscode: - docker: image: docker.cnb.cool/yichenyanyu/pytorch/pytorch2.7.0-cuda12.8-cudnn9-devel:latest