- Python 3.10.6
- Nvidia GPU
- Create venv
py -3.10 -m venv cuda
- Activate venv
cuda activate
- Install libs
pip install matplotlib numpy pylzma ipkernel jupyter
pip install torch --index-url https://download.pytorch.org/whl/cu118
- Install a new kernel for Jupyter Notebook
python -m ipykernel install --user --name=cuda --display-name "cuda-gpt"
- Start Jupyter Notebook
jupyter notebook