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

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:
 

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
+

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
+

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