File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
5687pip install diffct
5788```
89+ </details >
5890
5991## 📝 Citation
6092
You can’t perform that action at this time.
0 commit comments