amithkk / amithkk/stable-diffusion-k8s

support for h100

Open
#10 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Dockerfile
Stars
91
Forks
37
PR merge metrics
No merged PRs in 30d

Description

/root/miniconda3/lib/python3.8/site-packages/torch/cuda/__init__.py:145: UserWarning:
NVIDIA H100 80GB HBM3 with CUDA capability sm_90 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 compute_37.
If you want to use the NVIDIA H100 80GB HBM3 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.