Skip to content

Commit 87e879c

Browse files
committed
Update 1
1 parent a842eb0 commit 87e879c

File tree

1 file changed

+46
-1
lines changed

1 file changed

+46
-1
lines changed

README.md

Lines changed: 46 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,48 @@
11
# torchkan
22

3-
Under Development
3+
**Under Development**
4+
5+
6+
# TorchKAN: KAN Model Evaluation with PyTorch and CUDA
7+
8+
This project demonstrates the training, validation, and quantization of the simplified KAN model using PyTorch, with CUDA acceleration for improved performance. The project builds the `torchkan` library to create and evaluate KAN models on the MNIST dataset, as a preliminary test.
9+
10+
## Prerequisites
11+
12+
Before you begin, ensure you have the following installed on your system:
13+
- Python (version 3.6 or higher)
14+
- CUDA Toolkit (corresponding to the CUDA version required by your PyTorch installation)
15+
- cuDNN (compatible with your installed CUDA Toolkit)
16+
17+
## Installation (Tested on MacOS and Linux)
18+
19+
### 1. Clone the Repository
20+
21+
Start by cloning the repository containing the `torchkan` library and navigating into the project directory:
22+
23+
```bash
24+
git clone https://github.com/1ssb/torchkan.git
25+
cd torchkan
26+
pip install -r requirements.txt
27+
```
28+
29+
If not already installed:
30+
31+
```bash
32+
export PATH=/usr/local/cuda/bin:$PATH
33+
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
34+
```
35+
36+
To run the MNIST training, make sure you configure wandb and run the script:
37+
38+
```python
39+
python mnist.py
40+
```
41+
42+
## Contact
43+
44+
For questions contact me at Subhransu.Bhattacharjee@anu.edu.au
45+
46+
## Contributions
47+
48+
Contributions are welcome, please raise issues as required after commit Fin. This repository is still under development and testing. The code is licensed under the MIT License.

0 commit comments

Comments
 (0)