深度学习框架PyTorch安装

深度学习框架PyTorch安装

深度学习框架PyTorch安装

1、访问 https://pytorch.org/get-started/locally/

2、选择版本、操作系统、安装的平台、语言、计算平台。

3、cmd中执行:

使用国外源:用时超过24小时

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

使用国内源:用时10分钟左右

pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple/

4、安装完成。

5、查看PyTorch的版本信息。

import torch as t

print(t.__version__)

发表回复

您的电子邮箱地址不会被公开。