mirror of https://github.com/astral-sh/uv
33 lines
1.5 KiB
Bash
Executable File
33 lines
1.5 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -ex
|
|
|
|
cargo build
|
|
uv="$(pwd)/target/debug/uv"
|
|
unset VIRTUAL_ENV
|
|
export RUST_LOG=uv_distribution_types=debug,uv_distribution::distribution_database=debug
|
|
|
|
# 1. Single platform
|
|
${uv} venv --clear -q -p 3.13 && echo "torch" | ${uv} pip compile - --index https://download.pytorch.org/whl/test/variant --no-annotate --preview --refresh
|
|
|
|
export NV_VARIANT_PROVIDER_FORCE_CUDA_DRIVER_VERSION=12.8
|
|
export NV_VARIANT_PROVIDER_FORCE_SM_ARCH=9.0
|
|
${uv} venv --clear -q -p 3.13 && echo "torch" | ${uv} pip compile - --index https://download.pytorch.org/whl/test/variant --no-annotate --preview
|
|
unset NV_VARIANT_PROVIDER_FORCE_CUDA_DRIVER_VERSION
|
|
unset NV_VARIANT_PROVIDER_FORCE_SM_ARCH
|
|
|
|
# 2. Universal with lockfile
|
|
# Three cases: Sync (new/updated lock), Sync (fresh lock), Sync (noop)
|
|
( cd scripts/packages/torch_user && ${uv} venv -p 3.13 -c -q && rm -f uv.lock && ${uv} sync --preview )
|
|
( cd scripts/packages/torch_user && ${uv} venv -p 3.13 -c -q && ${uv} sync --preview )
|
|
( cd scripts/packages/torch_user && ${uv} sync --preview )
|
|
|
|
export NV_VARIANT_PROVIDER_FORCE_CUDA_DRIVER_VERSION=12.8
|
|
export NV_VARIANT_PROVIDER_FORCE_SM_ARCH=9.0
|
|
( cd scripts/packages/torch_user && ${uv} venv -p 3.13 -c -q && rm -f uv.lock && ${uv} sync --preview )
|
|
( cd scripts/packages/torch_user && ${uv} venv -p 3.13 -c -q && ${uv} sync --preview )
|
|
( cd scripts/packages/torch_user && ${uv} sync --preview )
|
|
unset NV_VARIANT_PROVIDER_FORCE_CUDA_DRIVER_VERSION
|
|
unset NV_VARIANT_PROVIDER_FORCE_SM_ARCH
|
|
|