updated installation for cuda 12.6

torchmd · Feb 11, 2025 · 1facb60 · 1facb60
1 parent 1deecd1
commit 1facb60
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/docs/source/installation.rst b/docs/source/installation.rst
@@ -54,8 +54,7 @@ It is recommended to install the same version as the one used by torch.
 
 .. code-block:: shell
 
-   conda install -c conda-forge cuda-nvcc cuda-libraries-dev cuda-version gxx pytorch=*=*cuda*
-
+   conda install -c conda-forge python=3.10 cuda-version=12.6 cuda-nvvm cuda-nvcc cuda-libraries-dev
 
 
 * CUDA<12