docs: add cuda-version note to conda install instructions#102
Conversation
The default `conda install nvmolkit` may select cuda-version=13, causing PyTorch to fall back to CPU-only on systems with CUDA 12.x drivers. Add a note instructing users to specify cuda-version to match their system.
|
| Filename | Overview |
|---|---|
| README.md | Adds a note warning users about potential CPU-only PyTorch installs when the default conda solve picks cuda-version=13, with a workaround command and a link to available PyTorch builds. The link URL uses a non-standard Anaconda path that may not work as intended, and a blank line is missing between the closing code fence and the following paragraph. |
Last reviewed commit: 6581e8e
The default `conda install nvmolkit` may select cuda-version=13, causing PyTorch to fall back to CPU-only on systems with CUDA 12.x drivers. Add a note instructing users to specify cuda-version to match their system.
89721d8 to
63fb66a
Compare
| conda install -c conda-forge nvmolkit | ||
| ``` | ||
|
|
||
| **Note**: If you are not running CUDA 13, the default install may select `cuda-version=13` and |
There was a problem hiding this comment.
I don't think this is exactly right. cuda-version should
- Be eq or less than your nvidia-smi
- Have a supported pytorch version in conda-forge
Suggest linking the torch page here and having the instructions say pick a cuda-version <= nvidia-smi driver version that has a matching torch major version that they want + the right cuda version.
Alternatively. Suggest they install torch first. Check that it's GPU enabled. Then install nvmolkit
There was a problem hiding this comment.
Added two install options: (A) specify cuda-version explicitly — must be ≤ nvidia-smi version with a matching PyTorch build on conda-forge (linked the builds page), and (B) install PyTorch with GPU support first, verify with torch.cuda.is_available(), then install nvmolkit.
There was a problem hiding this comment.
Isn't pytorch-cuda dead? It's from the pytorch channel which they no longer use. 12.4 was the last one they support. Current support is only through conda-forge
There was a problem hiding this comment.
Removed it and consolidated the two options into one since without pytorch-cuda, the "install torch first" approach will require specifying cuda-version explicitly, making it essentially the same flow as option A split into steps.
Add two options for ensuring GPU-accelerated PyTorch: specifying cuda-version explicitly with link to conda-forge PyTorch builds, or installing PyTorch first and verifying GPU support.
pytorch-cuda is from the old PyTorch channel (last at CUDA 12.4) and not available on conda-forge. Consolidate to a single option using explicit cuda-version.
The default
conda install nvmolkitmay select cuda-version=13, causing PyTorch to fall back to CPU-only on systems with CUDA 12.x drivers. Add a note instructing users to specify cuda-version to match their system.For addressing #101