pytorch安装到pip
时间: 2023-11-03 18:57:21 浏览: 141
升级pip到20.0.2+用本地安装包安装pytorch
PyTorch可以通过pip命令进行安装。你可以使用以下命令安装PyTorch:
pip3 install torch torchvision torchaudio
安装过程中,建议不要移动鼠标,以免导致下载停止,需要重新开始。整个过程大约需要十几分钟。安装完成后,你可以在编译器中进行测试。你可以使用以下代码测试是否成功安装PyTorch:
import torch
print(torch.__version__) # 输出版本号
x = torch.rand(5, 3) # 进行简单的计算测试
print(x)
print(torch.cuda.is_available()) # 测试是否支持CUDA,返回True表示支持,否则表示仅支持CPU。
如果你的编译器是PyCharm,我认为它比Spyder更好用。
阅读全文