diff --git a/docs/source/installation.rst b/docs/source/installation.rst index 9d7c04b9..3a956a5b 100644 --- a/docs/source/installation.rst +++ b/docs/source/installation.rst @@ -21,8 +21,7 @@ GPU-compatible version of PyTorch first. have the most up-to-date guidance for this. Currently, this is the command: .. code-block:: console - - $ pip install torch --index-url https://download.pytorch.org/whl/cu129 + $ pip install -r requirements-gpu.txt Then, install NAM using pip: diff --git a/environments/environment_cpu_apple.yml b/environments/environment_cpu_apple.yml index d0b6e9b0..bb1403f3 100644 --- a/environments/environment_cpu_apple.yml +++ b/environments/environment_cpu_apple.yml @@ -7,7 +7,6 @@ name: nam channels: - conda-forge # pytest-mock - - pytorch - defaults dependencies: - python>=3.9 @@ -24,7 +23,6 @@ dependencies: - pytest-mock # Performance note: # https://github.com/sdatkinson/neural-amp-modeler/issues/505 - - pytorch - semver - tensorboard - tqdm @@ -36,3 +34,5 @@ dependencies: - transformers>=4 # See requirements.txt - wavio >=0.0.5 - -e .. + - torch + diff --git a/environments/environment_gpu.yml b/environments/environment_gpu.yml index d3e91e0a..a50a2a70 100644 --- a/environments/environment_gpu.yml +++ b/environments/environment_gpu.yml @@ -5,7 +5,6 @@ name: nam channels: - conda-forge # pytest-mock - - pytorch - nvidia # GPU - defaults dependencies: @@ -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 @@ -37,3 +31,5 @@ dependencies: - transformers>=4 # See requirements.txt - wavio >=0.0.5 - -e .. + - torch + diff --git a/requirements-gpu.txt b/requirements-gpu.txt new file mode 100644 index 00000000..620b1d49 --- /dev/null +++ b/requirements-gpu.txt @@ -0,0 +1,5 @@ +# requirements-gpu.txt +# Install PyTorch with CUDA support for GPU acceleration +# Adjust the CUDA version (cu118) as needed for your system + +torch torchvision --index-url https://download.pytorch.org/whl/cu129