commit 64eb2538dbf9e57d324448f224c66c7fac918efb
parent fe997ace95a7c81d70f516658d1fff2c5ecb2e7a
Author: Hans P. Möller <[email protected]>
Date: Sun, 29 Sep 2024 01:08:18 -0300
update pytorch-cuda version (#478)
update pytorch-cuda version to match https://github.com/sdatkinson/neural-amp-modeler/pull/342
Diffstat:
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/docs/source/installation.rst b/docs/source/installation.rst
@@ -11,7 +11,7 @@ of PyTorch first:
.. code-block:: console
- $ conda install -y pytorch pytorch-cuda=11.8 -c pytorch -c nvidia
+ $ conda install -y pytorch pytorch-cuda=12.1 -c pytorch -c nvidia
Next, install NAM using pip: