Skip to content

minimum torch 2.2.2 #97

minimum torch 2.2.2

minimum torch 2.2.2 #97

Workflow file for this run

name: Build AutoAWQ Wheels with CUDA
on:
push:
branches:
- "downgrade_build_torch"
jobs:
# release:
# # Retrieve tag and create release
# name: Create Release
# runs-on: ubuntu-latest
# outputs:
# upload_url: ${{ steps.create_release.outputs.upload_url }}
# steps:
# - name: Checkout
# uses: actions/checkout@v3
# - name: Extract branch info
# shell: bash
# run: |
# echo "release_tag=${GITHUB_REF#refs/*/}" >> $GITHUB_ENV
# - name: Create Release
# id: create_release
# uses: "actions/github-script@v6"
# env:
# RELEASE_TAG: ${{ env.release_tag }}
# with:
# github-token: "${{ secrets.GITHUB_TOKEN }}"
# script: |
# const script = require('.github/workflows/scripts/github_create_release.js')
# await script(github, context, core)
build_wheels:
name: Build AWQ
runs-on: ${{ matrix.os }}
# needs: release
strategy:
matrix:
os: [ubuntu-20.04]
pyver: ["3.9"]
defaults:
run:
shell: pwsh
env:
MIN_TORCH_VER: "2.2.0"
steps:
- name: Free Disk Space
uses: jlumbroso/free-disk-space@v1.3.0
if: runner.os == 'Linux'
with:
tool-cache: false
android: true
dotnet: true
haskell: true
large-packages: false
docker-images: true
swap-storage: false
- uses: actions/checkout@v3
- uses: actions/setup-python@v3
with:
python-version: ${{ matrix.pyver }}
- name: Setup Mamba
uses: conda-incubator/setup-miniconda@v2.2.0
with:
activate-environment: "build"
python-version: ${{ matrix.pyver }}
miniforge-variant: Mambaforge
miniforge-version: latest
use-mamba: true
add-pip-as-python-dependency: true
auto-activate-base: false
- name: Install Dependencies
run: |
# Install torch
python -m pip install --upgrade --no-cache-dir torch==$env:MIN_TORCH_VER
python -m pip install build setuptools wheel
# Print version information
python --version
python -c "import torch; print('PyTorch:', torch.__version__)"
- name: Build Wheel
run: |
python setup.py sdist bdist_wheel
# - name: Upload Assets
# uses: shogo82148/actions-upload-release-asset@v1
# with:
# upload_url: ${{ needs.release.outputs.upload_url }}
# asset_path: ./dist/*.whl