526互联

pytorch安装

发布时间 2023-04-07 00:07:01作者: 周XX

pytorch的安装就是做选择题

  1. 选择版本。版本主要分为稳定版和尝鲜版。对于大多数用户来说，稳定版就好啦，稳定版是经过充分测试验证的；但尝鲜版就不一样了，很多特性并不稳定。
  2. 选择系统。
  3. 选择安装工具。比如conda，pip。笔者通常都会用pip
  4. 选择开发语言。这个系列用python
  5. 选择计算框架。有GPU就选择CUDA，没有就选CPU。

按照笔者的选择，使用的安装语句是

pip3 install torch torchvision torchaudio

安装完成就可以试试看了

import torch
x = torch.rand(5, 3)
print(x)

如果上面的语句能正常打印就算安装好了

参考：https://pytorch.org/get-started/locally/#windows-pip

返回目录