Skip to content

Instantly share code, notes, and snippets.

@saforem2
Last active April 24, 2025 19:00
Show Gist options
  • Select an option

  • Save saforem2/d61c12eb6e5c78affbd82d5de05e2d16 to your computer and use it in GitHub Desktop.

Select an option

Save saforem2/d61c12eb6e5c78affbd82d5de05e2d16 to your computer and use it in GitHub Desktop.

Installing PyTorch 2.7.0 on Aurora with uv

#[🐍 aurora_nre_models_frameworks-2024.2.1_u1]
#[01:06:45 PM][x4001c3s0b0n0][/flare/datascience/foremans/projects/saforem2/tmp/2025-04-24-130620]
; uv venv --python=$(which python3) --system-site-packages # "venvs/$(echo "${CONDA_PREFIX}" | tr "\/" " " | awk '{print $NF}')"
Using CPython 3.10.14 interpreter at: /opt/aurora/24.180.3/frameworks/aurora_nre_models_frameworks-2024.2.1_u1/bin/python3
Creating virtual environment at: .venv
Activate with: source .venv/bin/activate
#[🐍 aurora_nre_models_frameworks-2024.2.1_u1]
#[01:06:51 PM][x4001c3s0b0n0][/flare/datascience/foremans/projects/saforem2/tmp/2025-04-24-130620]
; source .venv/bin/activate
#[🐍 aurora_nre_models_frameworks-2024.2.1_u1](πŸ‘» 2025-04-24-130620)
#[01:07:03 PM][x4001c3s0b0n0][/flare/datascience/foremans/projects/saforem2/tmp/2025-04-24-130620]
; uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/xpu
Resolved 22 packages in 801ms
Prepared 19 packages in 43.24s
β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘β–‘ [0/22] Installing wheels...
         warning: Failed to hardlink files; falling back to full copy. This may lead to degraded performance.
         If the cache and target directories are on different filesystems, hardlinking may not be supported.
         If this is intentional, set `export UV_LINK_MODE=copy` or use `--link-mode=copy` to suppress this warning.
Installed 22 packages in 7m 56s
 + filelock==3.13.1
 + fsspec==2024.6.1
 + intel-cmplr-lib-rt==2025.0.4
 + intel-cmplr-lib-ur==2025.0.4
 + intel-cmplr-lic-rt==2025.0.4
 + intel-pti==0.10.1
 + intel-sycl-rt==2025.0.4
 + jinja2==3.1.4
 + markupsafe==2.1.5
 + mpmath==1.3.0
 + networkx==3.3
 + numpy==2.1.2
 + pillow==11.0.0
 + pytorch-triton-xpu==3.3.0
 + setuptools==70.2.0
 + sympy==1.13.3
 + tcmlib==1.2.0
 + torch==2.7.0+xpu
 + torchaudio==2.7.0+xpu
 + torchvision==0.22.0+xpu
 + typing-extensions==4.12.2
 + umf==0.9.1
took: 0h:08m:41s
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment