Skip to content

ISSUE 534: Deprecate conda environment.yml / move to pip #552

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 9 commits into
base: main
Choose a base branch
from
Open

ISSUE 534: Deprecate conda environment.yml / move to pip #552

Show file tree
Hide file tree
Changes from 5 commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
9878ae5
use pip to install pytorch
HundredBillion Apr 26, 2025
80322a2
update documentation
HundredBillion Apr 26, 2025
856d73e
fix documentation grammer
HundredBillion Apr 26, 2025
3c62ad4
update installation doc
HundredBillion Apr 28, 2025
2dc6316
update instruction to install pytorch via pip
HundredBillion Apr 28, 2025
b89452b
make PR change request
HundredBillion Jun 25, 2025
5d42b65
create a new file so that CUDA users can still enable CUDA-accelerate…
HundredBillion Jun 25, 2025
1e1baed
update file to install pytorch with cuda
HundredBillion Sep 17, 2025
2fb75dd
Merge branch 'main' into issue-534
HundredBillion Sep 17, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 11 additions & 5 deletions docs/source/installation.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,14 +15,20 @@ Step 2: Install NAM

Now that we have Miniconda, we can install NAM using it.

(Windows / Linux users) If your computer has an nVIDIA GPU, you should install a
GPU-compatible version of PyTorch first:
.. note::

.. code-block:: console
(Windows / Linux users) If your computer has an nVIDIA GPU, you should install a
GPU-compatible version of PyTorch first.

.. code-block:: console

$ pip3 install torch torchvision torchaudio \
--index-url https://download.pytorch.org/whl/cu118

$ conda install -y pytorch pytorch-cuda=12.1 -c pytorch -c nvidia
please refer to the official PyTorch website:
https://pytorch.org/get-started/locally/ for the latest instructions.

Finally, install NAM using pip:
Install NAM using pip:

.. code-block:: console

Expand Down
5 changes: 3 additions & 2 deletions environments/environment_cpu_apple.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@
name: nam
channels:
- conda-forge # pytest-mock
- pytorch
- defaults
dependencies:
- python>=3.9
Expand All @@ -24,7 +23,6 @@ dependencies:
- pytest-mock
# Performance note:
# https://github.com/sdatkinson/neural-amp-modeler/issues/505
- pytorch
- semver
- tensorboard
- tqdm
Expand All @@ -36,3 +34,6 @@ dependencies:
- transformers>=4 # See requirements.txt
- wavio >=0.0.5
- -e ..
- torch
- torchvision
- torchaudio
9 changes: 3 additions & 6 deletions environments/environment_gpu.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
name: nam
channels:
- conda-forge # pytest-mock
- pytorch
- nvidia # GPU
- defaults
dependencies:
Expand All @@ -21,11 +20,6 @@ dependencies:
- pydantic>=2
- pytest
- pytest-mock
- pytorch::pytorch
# If your GPU isn't being detected, you may need a different version.
# You're going to need to look at Table 3 here:
# https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cuda-major-component-versions__table-cuda-toolkit-driver-versions
- pytorch::pytorch-cuda=12.1 # GPU
- semver
- tensorboard
- tqdm
Expand All @@ -37,3 +31,6 @@ dependencies:
- transformers>=4 # See requirements.txt
- wavio >=0.0.5
- -e ..
- torch
- torchvision
- torchaudio