logo
0
1
Login

Pytorch CUDA 基础镜像

本项目为 pytorchcuda 的基础镜像,可以作为基础docker进行相关AI项目的开发。

已有镜像列表

镜像地址Python版本Pytorch版本CUDA版本
docker.cnb.cool/yichenyanyu/pytorch/pytorch2.6.0-cuda12.6-cudnn9-devel:latestPython BadgePytorch Badgecuda Badge
docker.cnb.cool/yichenyanyu/pytorch/pytorch2.68.0-cuda12.9-cudnn9-devel:latestPython BadgePytorch Badgecuda Badge

使用说明

根据需要在自己的项目 .cnb.yml 文件中引用当前项目的镜像,例如:

$: vscode: - docker: image: docker.cnb.cool/yichenyanyu/pytorch/pytorch2.7.0-cuda12.8-cudnn9-devel:latest

About

No description, topics, or website provided.
Language
Shell80.6%
Dockerfile10%
Markdown9.5%