Pip安装Pytorch gpu版

1、安装指令(注意需要指定torch的版本,不然会下载一堆不同的版本)

pip3 install torch==1.13.0 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

2、安装好后测试CUDA是否可用

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

你可能感兴趣的:(python,pytorch,pip,深度学习)