replicate/cog

Pick latest CUDA patch version when specifying minor version

Fermée

#813 ouverte le 27 nov. 2022

 (2 commentaires) (0 réaction) (0 personne assignée)Python (356 forks)batch import
good first issuenexttype/bug

Métriques du dépôt

Stars
 (5 724 étoiles)
Métriques de merge PR
 (Merge moyen 7j 8h) (19 PRs mergées en 30 j)

Description

build:
  python_version: "3.10"
  cuda: "11.6"
  python_packages:
   - "torch==1.13.0"

This automatically selects CUDA 11.6.0 for the base image. It should pick the latest patch version and use 11.6.2.

Guide contributeur