add installation notes for CUDA 12.4, pyenv, and PyTorch; include useful links

This commit is contained in:
Joseph Hopfmüller
2024-11-16 00:03:43 +01:00
parent 2bf47dc0c0
commit 939a511625
4 changed files with 149 additions and 0 deletions

25
notes/pytorch-install.md Normal file
View File

@@ -0,0 +1,25 @@
# pytorch install
## create venv
```bash
python -m venv ./.venv
source ./.venv/bin/activate
```
## install pytorch
> https://pytorch.org/get-started/locally/
```bash
pip install torch torchvision torchaudio
```
## test pytorch install
```python
import torch
x = torch.rand(5, 3)
print(x)
torch.cuda.is_available()
```