Skip to content

PyTorch

Nimrod Mesika edited this page Jul 14, 2024 · 2 revisions
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
>>> import torch
>>> torch.cuda.is_available()
True
>>>

Clone this wiki locally