PyTorch - nimrody/knowledgebase GitHub Wiki

• Installing pytorch on WSL2 with Cuda

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

>>> import torch
>>> torch.cuda.is_available()
True
>>>