Skip to content

Commit bb3798d

Browse files
committed
Add installation instructions for Conda users with CUDA 12 and 11 support
1 parent 37bb049 commit bb3798d

1 file changed

Lines changed: 32 additions & 0 deletions

File tree

README.md

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,9 +52,41 @@ diffct/
5252

5353
### Installation
5454

55+
To get started, create a Conda environment and install the necessary CUDA toolkit and PyTorch version.
56+
57+
<details>
58+
<summary><b>🐍 For Conda Users</b></summary>
59+
60+
**CUDA 12:**
61+
```bash
62+
# Create and activate conda environment
63+
conda create -n diffct python=3.12
64+
conda activate diffct
65+
66+
# Install CUDA Toolkit, PyTorch, and Numba
67+
conda install nvidia/label/cuda-12.8.1::cuda-toolkit
68+
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
69+
pip install numba-cuda[cu12]
70+
71+
# Install diffct
72+
pip install diffct
73+
```
74+
75+
**CUDA 11:**
5576
```bash
77+
# Create and activate conda environment
78+
conda create -n diffct python=3.12
79+
conda activate diffct
80+
81+
# Install CUDA Toolkit, PyTorch, and Numba
82+
conda install nvidia/label/cuda-11.8.0::cuda-toolkit
83+
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
84+
pip install numba-cuda[cu11]
85+
86+
# Install diffct
5687
pip install diffct
5788
```
89+
</details>
5890

5991
## 📝 Citation
6092

0 commit comments

Comments
 (0)