DeepSpeed
Specify torchvision in nv-ds-chat workflow (prevents errors with torch 2.6)
#6982
Merged
Overview
Commits
5
Changes
View On GitHub

Specify torchvision in nv-ds-chat workflow (prevents errors with torch 2.6) #6982

loadams merged 5 commits into master from loadams/ds-chat-fixes
loadams
loadams Pin torch in nv-ds-chat workflow
6d57a4a2
loadams loadams requested a review from tjruwase tjruwase 317 days ago
loadams Turn off NCCL debug
2c8488b7
tjruwase
tjruwase approved these changes on 2025-01-30
loadams Ensure we get torch+cu vs torch+cpu
289871fd
loadams Merge branch 'master' into loadams/ds-chat-fixes
63097c44
loadams Add torchvision at same install time as torch
634b29ec
loadams loadams changed the title Pin torch in nv-ds-chat workflow Specify torchvision in nv-ds-chat workflow (prevents errors with torch 2.6) 316 days ago
loadams loadams enabled auto-merge 316 days ago
loadams loadams merged c963c21f into master 316 days ago
loadams loadams deleted the loadams/ds-chat-fixes branch 316 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum