diffsbdd

Community

Seamlessly migrate diffusion-based drug design models to Ascend NPU.

Authordongg622
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This Skill streamlines the process of migrating diffusion model projects to Ascend hardware, reducing setup complexity and compatibility issues.

Core Features & Use Cases

  • Environment Setup: Guides users through cloning repositories, creating Conda environments, and configuring CANN for Ascend.
  • Dependency Management: Automates installation of compatible PyTorch, torch_npu, and necessary libraries, including source compilation of torch_scatter.
  • Code Adaptation: Injects transfer routines to redirect CUDA calls to NPU seamlessly for model inference and training tasks.
  • Validation & Verification: Provides scripts to verify environment readiness and inference correctness on Ascend hardware.

Quick Start

Follow the instructions to clone the repository, set up the environment, compile necessary dependencies, adapt code with transfer_to_npu, and run inference validation.

Dependency Matrix

Required Modules

None required

Components

scriptsreferences

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: diffsbdd
Download link: https://github.com/dongg622/china-ai-chip-skill/archive/main.zip#diffsbdd

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.