Skip to content
Open
Show file tree
Hide file tree
Changes from 3 commits
Commits
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
10 changes: 4 additions & 6 deletions docs/source/installation.rst
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,11 @@ 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:
GPU-compatible version of PyTorch first. For the recommended installation
commands, please refer to the official PyTorch website:
https://pytorch.org/get-started/locally/

.. code-block:: console

$ conda install -y pytorch pytorch-cuda=12.1 -c pytorch -c nvidia

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

.. code-block:: console

Expand Down
5 changes: 1 addition & 4 deletions environments/environment_cpu_apple.yml
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 @@ -22,9 +21,6 @@ dependencies:
- pydantic
- pytest
- pytest-mock
# Performance note:
# https://github.com/sdatkinson/neural-amp-modeler/issues/505
- pytorch
- semver
- tensorboard
- tqdm
Expand All @@ -36,3 +32,4 @@ dependencies:
- transformers>=4 # See requirements.txt
- wavio >=0.0.5
- -e ..
- torch
7 changes: 1 addition & 6 deletions environments/environment_gpu.yml
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,4 @@ dependencies:
- transformers>=4 # See requirements.txt
- wavio >=0.0.5
- -e ..
- torch