*Pyorch官网
*Pytorch安装指南
安装Pytorch的pip命令
pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
对应的库安装完,可以用python代码验证PyTorch是否安装成功
import torch
torch.cuda.is_available()
网友评论